[Algorithm] 가장 가까운 같은 글

 

문제 설명

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

 

문제 유형

• 가장 가까운 이전 위치 찾기 (Closest Previous Index)

 

아이디어

• 문자열 배열을 순회하며 현재 알파벳 위치를 저장한다.
• 이전 알파벳 위치가 없다면 결과값에 -1을 저장한다.
• 이전 알파벳 위치가 있다면 결과값에 현재 알파벳 위치를 계산하여 결과값에 저장한다.

 

의사코드

1. 알파벳 위치를 저장할 변수(alphaPos)를 선언한다.
2. alphaPos의 모든 요소의 값을 -1로 초기화한다.
3. [반복문] 문자열의 크기만큼 반복한다.
   1. alphaPos의 인덱스 값을 구한다. (s[i] - 'a')
   2. alphaPos의 인덱스 위치값이 -1이라면 결과값에 -1을 저장한다.
   3. alphaPos의 인덱스 위치값이 -1이 아니라면 알파벳 위치를 계산하여 결과값에 저장한다.
   4. alphaPos에 현재 알파벳 위치를 저장한다.
4. 결과값(answer)을 반환한다.

 

코드 작성

#include <string>
#include <vector>

using namespace std;

int alphaPos[26];

vector<int> solution(string s) {
    vector<int> answer;
    std::fill(std::begin(alphaPos), std::end(alphaPos), -1);
    
    for (int i = 0; i < s.size(); i++)
    {
        int idx = s[i] - 'a';
        
        if (alphaPos[idx] == -1) answer.push_back(-1);
        else answer.push_back(i - alphaPos[idx]);
        
        alphaPos[idx] = i;
    }
    
    return answer;
}