프로그래머스 가장 가까운 글자 - 자바 (java) HashMap

쉬워 보였는디.. 헷갈리네

 

문제를 읽어보자.

자신의 앞에 자신과 동일한 알파벳이 존재하는 경우 그 간격을 출력하면 되고, 없을경우 -1을 출력하면 된다.

위치 인덱스를 저장해야하므로 HashMap을 사용해서 value랑 key값으로 처리하면 되겠다! 라는 생각이 들었다.

 

물론 s를 char[]에 담은 후, 이중 반복문으로 같은 알파벳이 나오면 거리를 구해서 답을 구할 수도 있다.

 

import java.util.*;
 
class Solution {
    public int[] solution(String s) {
        int[] answer = new int[s.length()];
        HashMap<Character, Integer> map = new HashMap<>(); 
        
        for (int i = 0; i < s.length(); i++) {
            char c = s.charAt(i);
            if (map.containsKey(c)) { 
                int prevIdx = map.get(c);
                answer[i] = i - prevIdx; 
            }
            map.put(c, i); 
        }
        
        for (int i = 0; i < answer.length; i++) {
            if (answer[i] == 0) {
                answer[i] = -1;
            }
        }
        
        return answer;
    }
}

 

import java.util.*;

class Solution {
    public int[] solution(String s) {
        int[] answer = new int[s.length()];
        HashMap<Character, Integer> map = new HashMap<>();  // HashMap으로 value와 key 연결.
        
        for (int i = 0; i < s.length(); i++) {
            char c = s.charAt(i);
            if (map.containsKey(c)) {                                                  //저장된 키와 같은경우
                int prevIdx = map.get(c);
                answer[i] = i - prevIdx;                                                 // 현재 위치와 이전 위치의 거리 저장
            }
            map.put(c, i);                                                                    // 현재 문자와 위치를 키와 벨류로 저장
        }
        
        for (int i = 0; i < answer.length; i++) {
            if (answer[i] == 0) {                                                            // 이전에 등장한 적 없는 문자일 경우
                answer[i] = -1;
            }
        }
        
        return answer;
    }
}