가장 가까운 같은 글자/C++

문제 설명

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

풀이 방법

1. map 자료형 m을 선언한다.

2. 처음 나온 문자일 시 answer에 -1을 담고 m에 문자와 해당 문자의 인덱스를 담는다.

3. 처음 나온 문자가 아닐 시 answer에 현재 문자의 인덱스에 m에 저장된 같은 문자를 key로 가지는 value를 뺀 값을 담는다.

4. m에서 해당 key값의 value를 현재 문자의 인덱스로 수정한다.

5. answer 리턴

#include <string>
#include <vector>
#include <map>

using namespace std;

vector<int> solution(string s) {
    vector<int> answer;
    
    map<char,int> m;
    
    for(int i = 0; i < s.length(); i++){
        if(m.count(s[i]) == 0) {
            m.insert(pair<char,int> (s[i], i));
            answer.push_back(-1);
        }
        else{
            answer.push_back(i - m[s[i]]);
            m[s[i]] = i;
        }
    }
    
    return answer;
}