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

2023. 10. 20. 11:54Algorithm

문제 설명 

문자열 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 result 
"banana" [-1, -1, -1, 2, 2, 2] 
"foobar"  [-1, -1, 1, -1, -1, -1]

 

정답
import java.util.ArrayList;

class Solution {
    public int[] solution(String s) {
        int n = s.length();
        int[] answer = new int[n];
        ArrayList<Integer>[] indexes = new ArrayList[26];

        for (int i=0; i<26; i++) {
            indexes[i] = new ArrayList<>();
        }

        for (int i=0; i<n; i++) {
            int index = s.charAt(i) - 'a';
            if (indexes[index].isEmpty()) {
                answer[i] = -1;
            } else {
                int closest = indexes[index].get(indexes[index].size() - 1);
                answer[i] = i - closest;
            }
            indexes[index].add(i);
        }

        return answer;
    }
}

indexes -> 0~26을 아스키코드 형태로 A~Z를 나타낸 것이고 해당 값이 나오면 최신 인덱스 값을 갱신해 비교를 위해 사용한다.

index의 경우 아스키코드의 값을 구해 비교하고 if 문에서 비교를 통해 이전에 값이 나오지 않았다면 answer에 값을 -1을 넣어주고 아니라면 최신 값을 갱신해서 인덱스에 넣어주면 된다

 

해당 코드말고도 정답을 보면 HashMap을 이용해 푸는 경우도 있는데 해당 경우도 생각해보는게 좋았을거 같다