[프로그래머스] Lv.1 가장 가까운 같은 글자 - JAVA

문제

문자열 s가 주어졌을 때, s의 각 위치마다 자신보다 앞에 나왔으면서, 자신과 가장 가까운 곳에 있는 같은 글자가 어디 있는지 알고 싶습니다.
예를 들어, s="banana"라고 할 때,  각 글자들을 왼쪽부터 오른쪽으로 읽어 나가면서 다음과 같이 진행할 수 있습니다.

b는 처음 나왔기 때문에 자신의 앞에 같은 글자가 없습니다. 이는 -1로 표현합니다.
a는 처음 나왔기 때문에 자신의 앞에 같은 글자가 없습니다. 이는 -1로 표현합니다.
n은 처음 나왔기 때문에 자신의 앞에 같은 글자가 없습니다. 이는 -1로 표현합니다.
a는 자신보다 두 칸 앞에 a가 있습니다. 이는 2로 표현합니다.
n도 자신보다 두 칸 앞에 n이 있습니다. 이는 2로 표현합니다.
a는 자신보다 두 칸, 네 칸 앞에 a가 있습니다. 이 중 가까운 것은 두 칸 앞이고, 이는 2로 표현합니다.
따라서 최종 결과물은 [-1, -1, -1, 2, 2, 2]가 됩니다.

문자열 s이 주어질 때, 위와 같이 정의된 연산을 수행하는 함수 solution을 완성해주세요.

 

 

제한사항

1 ≤ s의 길이 ≤ 10,000
	● s은 영어 소문자로만 이루어져 있습니다.

 

 

입출력 예시

 

 

풀이

s를 길이만큼 반복하여 index의 각 글자를 map에서 있는지 확인한다.

null일 경우에는 map에 현재 index를 넣고 result 배열에는 -1을 담는다.

null이 아닌 경우에는 현재 index와 map에서 나온 index를 빼서 배열에 담고, map을 현재 index로 교체한다.

import java.util.*;

class Solution {
    public int[] solution(String s) {
        int[] answer = new int[s.length()];

        Map<Character, Integer> map = new HashMap<>();

        for(int i = 0; i < s.length(); i++) {
            char c = s.charAt(i);
            if(map.get(c) != null) {
                int index = map.get(c);
                answer[i] = i - index;
                map.put(c, i);
            } else {
                answer[i] = -1;
                map.put(c, i);
            }
        }

        return answer;
    }
}

 

 

문의사항이나 피드백은 댓글로 남겨주세요.