가장 가까운 같은 글자 - 코드카타

가장 가까운 같은 글자

문제 설명

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

 

---

설계

• key 값을 char / value 값을 자리 번호를 저장하는 map<char, int> CharNum 변수 생성
• 자리번호를 카운팅하는 int Count = 0; 생성
• for each로 파라미터 string 을 순회
• string 원자 char 값을 CharNum에서 find로 찾아보고 없으면 answer에 -1 push_back
• string 원자 char 값을 CharNum에서 find로 찾아보고 있으면 찾은 Key값의 Value(자리번호)와 비교해 answer에 push_back

구현

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <map>

using namespace std;

vector<int> solution(string s) {
    vector<int> answer;
    map<char, int> CharNum;
    int count = 0;
    for (char c : s)
    {
        map<char, int>::iterator it = CharNum.find(c);
        if (it == CharNum.end())
        {
            CharNum.emplace(pair<char, int>(c, count));
            answer.push_back(-1);
        }
        else
        {
            answer.push_back(count - it->second);
            CharNum[c] = count;
        }

        count++;
    }
    return answer;
}

---

리팩토링

 

지금 내가 구현한 코드는 모든 문자를 입력 받는다면 모를까 지금의 문제는 소문자로 제한사항이 나온 상태 임으로 map에 저장하지말고 vector<int>로 소문자 26개만 순서대로 자리만 저장 하면 더 나은 코드가 될것 같다.

#include <string>
#include <vector>

using namespace std;

vector<int> solution(string s) {
    vector<int> answer;

    // ASCII 코드는 0~255까지 있으므로 256 크기의 배열을 생성합니다.
    // -1로 초기화하여 해당 문자가 아직 등장하지 않았음을 표시합니다.
    // 이렇게 하면 소문자('a')뿐만 아니라 대문자, 특수문자도 처리 가능합니다.
    vector<int> last_pos(256, -1); 

    // 문자열 전체를 한 번 순회합니다. (시간 복잡도: O(N))
    for (int i = 0; i < s.length(); i++) {
        // 현재 문자(s[i])를 ASCII 숫자로 자동 변환하여 인덱스로 사용합니다.
        char current_char = s[i];

        // 1. 이 문자가 이전에 등장한 적이 없는 경우
        if (last_pos[current_char] == -1) {
            answer.push_back(-1);
        }
        // 2. 이 문자가 이전에 등장한 적이 있는 경우
        else {
            // (현재 위치 - 마지막 등장 위치)를 계산하여 저장
            answer.push_back(i - last_pos[current_char]);
        }

        // 현재 문자의 위치를 최신 위치(i)로 갱신합니다.
        last_pos[current_char] = i;
    }

    return answer;
}