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

문제

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

---

 

먼저 alphabet 배열을 생성하여 -1로 초기화 해준다.

alphabet배열은 등장하는 알바펫의 가장 최신 위치를 갱신해주는 역할이다.

이후 문자열 s를 순서대로 탐색하며 i번째 알파벳에 대한 answer[i]과 alphabet[s[i] - 'a']을 갱신해준다.

 

문제의 예시 "banana"로 동작 과정을 살펴보자.

i = 0 -> b : 처음 등장하는 알파벳이므로 answer[0]은 그대로 -1로 두고 alphabet[ 'b' - 'a']만 i로 갱신시켜준다.

alphabet = [-1, 0, -1, ..., -1], answer = [-1, -1, -1, -1, -1, -1]

i = 1 -> a : 처음 등장하는 알파벳이므로 answer[1]은 그대로 -1로 두고 alphabet[ 'a' - 'a']만 i로 갱신시켜준다.

alphabet = [1, 0, -1, ..., -1], answer = [-1, -1, -1, -1, -1, -1]

i = 2 -> n : 처음 등장하는 알파벳이므로 answer[2]은 그대로 -1로 두고 alphabet[ 'n' - 'a']만 i로 갱신시켜준다.

alphabet = [1, 0, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, 2, ... , -1], answer = [-1, -1, -1, -1, -1, -1]

i = 3 -> a : 이미 등장한 알바펫이므로

answer[3]은 i - alphabet['a' - 'a'] = 3 - 1  = 2로 갱신하고 alphabet[ 'a' - 'a']는 i로 갱신시켜준다.

alphabet = [3, 0, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, 2, ... , -1], answer = [-1, -1, -1, 2, -1, -1]

i = 4 -> n : 이미 등장한 알바벳이므로

answer[4]은 i - alphabet['n' - 'a'] = 4 - 2 = 2로 갱신하고 alphabet['n' - 'a']는 i로 갱신시켜준다.

alphabet = [3, 0, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, 4, ... , -1], answer = [-1, -1, -1, 2, 2, -1]

i = 5 -> a : 이미 등장한 알바펫이므로

answer[5]은 i - alphabet['a' - 'a'] = 5 - 3 = 2로 갱신하고 alphabet['a' - 'a']는 i로 갱신시켜준다.

alphabet = [5, 0, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, 4, ... , -1], answer = [-1, -1, -1, 2, 2, 2]

 

위와 로직으로 O(n) time으로 문제를 해결할 수 있었다.

#include <string>
#include <vector>

using namespace std;


vector<int> solution(string s) {
    vector<int> answer(s.length(), -1);
    vector<int> alphabet(27, -1);
    for (int i = 0; i < s.length(); i++) {
        int idx = s[i] - 'a';
        if (alphabet[idx] == -1) { //처음 등장한 알바펫이면
            alphabet[idx] = i;
            continue;
        }
        //이미 등장한 알파벳이면
        answer[i] = i - alphabet[idx];
        alphabet[idx] = i;
    }
    return answer;
}