문제풀이/프로그래머스

[JAVA , Lv.1] 가장 가까운 같은 글자

PhoB 2022. 12. 31. 18:10

[문제]

문자열 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을 완성해주세요.

입출력 예

 

S result
"banana" [-1, -1, -1, 2, 2, 2]
"foobar" [-1, -1, 1, -1, -1, -1]


[제한사항]

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


[접근]

 

처음에 주어지는 배열과 한번 나온 문자를 저장하는 배열 , 총 2개의 배열을 사용하였다.

두번째 배열과 계속해서 비교해가며 숫자를 카운트 하였다.



[내 풀이]

class Solution {
    public int[] solution(String s) {
        int[] answer = new int[s.length()];
        char[] c = s.toCharArray(); //문자 배열 생성
        char[] c2 = new char[s.length()];//한번 나온 문자를 저장
        
        
        for(int i=0; i<c.length; i++){
            int cnt =0;
            if(i==0) {
                answer[0] = -1;
                continue;
            }//맨처음 글자는 항상-1
            int k=i-1;
            //두번째 글자부터 검사
            while(true){
                if(c[i] == c[k]){//찾으면 cnt++하고 그자리에서 break;
                    cnt++;
                    break;
                }
                else cnt++;
                k--;
                if(k==-1) { //끝까지 돌았는데 못찾았다면.
                    cnt=0;
                    break;
                }
                
            }
            if(cnt >0) answer[i] = cnt;
            else answer[i] = -1;
        }
        return answer;
    }
}

[개선점]

HashMap을 사용하면 containsKey()를 사용하여 검사를 좀더 쉽게 할 수 있다.


[개선 코드]

class Solution {
    public int[] solution(String s) {
        int[] answer = new int[s.length()];
        
        HashMap<Character,Integer> hmap = new HashMap<>();
        for(int i=0; i<s.length(); i++){
            if(!hmap.containsKey(s.charAt(i))){
                //해시맵에 i번재 문자가 포함되어있지 않을때(처음나왔을 때도 포함)
                answer[i] = -1;
                hmap.put(s.charAt(i),i);//해시맵에 추가
            }
            else{
                int cnt = hmap.get(s.charAt(i));
                answer[i] = i - cnt;
                hmap.put(s.charAt(i), i);
            }
        }
        return answer;
    }
}