[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은 영어 소문자로만 이루어져 있습니다.

 

import java.util.HashMap;

class Solution {
    public int[] solution(String s) {
        int[] answer = new int[s.length()];
        HashMap<Character, Integer> prevStr = new HashMap<Character, Integer>();
        
        int idx = 0;
        for(int i = 0; i < s.length(); i++) {
            char ch = s.charAt(i);
            
            if(prevStr.get(ch) == null) {
                answer[idx++] = -1;
                prevStr.put(ch, i+1);
            }
            else {
                answer[idx++] = (i+1) - prevStr.get(ch);
                prevStr.put(ch, i+1);
            }
        }
        
        return answer;
    }
}

처음에는 문제를 잘못 이해하고 '자신과 가장 가까운 곳에 있는 같은 글자가 어디 있는지' 라는 뜻을 인덱스 위치로 생각했다.

 

금방 알아차렸지만, 만약 테스트 케이스는 다 통과하게끔 되어있고 실제 제출 시 사용되는 테스트 케이스들은 그렇지 않은 케이스였다면, 내가 작성한 코드는 0점이 되었을 것이다.

 

앞으로 좀 더 확실하게 문제를 이해하고 해결하는 습관을 가져야겠다.

 

 

출처

https://school.programmers.co.kr/learn/courses/30/lessons/142086