[프로그래머스] - 가장 가까운 같은 글자(C/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은 영어 소문자로만 이루어져 있습니다.

---

입출력 예

s	result
"banana"	[-1, -1, -1, 2, 2, 2]
"foobar"	[-1, -1, 1, -1, -1, -1]

---

[문제 풀이]

문제 풀이는 간단하다.

왼쪽에서부터 읽어나가는데 만약 같은 글자가 없다면 -1

있다면 가장 가까운 같은 글자에서 현재 글자까지의 인덱스 차이를 vector에 넣어주면 된다.

 

문제를 풀기 위해 각 알파벳의 마지막 위치를 저장할 vector를 alphabetIndex라는 이름으로 선언하고, -1로 초기화해준다.

a~z => (0~25), 각각의 인덱스는 해당 알파벳의 마지막 위치를 저장한다.

 

반복문을 통해 문자열 s를 순회하면서 아래의 과정을 거친다.

현재 문자에서 'a'를 빼면 0~25의 숫자를 얻을 수 있다. 이것을 alphabetIndex vector의 인덱스로 넣어주면 해당 알파벳의 마지막 위치를 알 수 있다.

 

 

1. 만약 알파벳의 마지막 위치가 -1이라면 이전에 한번도 나오지 않았으므로, answer 벡터에 -1를 넣어준다.

 

2. 만약 알파벳의 마지막 위치가 -1이 아니라면 이전에 저장된 위치일 것이므로, 현재 위치에서 저장된 위치를 빼준 값을 answer 벡터에 넣어준다.

 

3. 매 반복문에서 현재 알파벳의 위치를 현재 인덱스로 업데이트 해준다.

 

4. 반복문이 끝날때까지 1~3의 과정을 반복한다.

 

위 로직을 코드로 표현하면 아래와 같다.

 

[코드]

[GitHub]

GitHub - EGyeom/ProblemSolving

코드가 깨끗하고 명확하게 보일진 모르겠지만, 누군가에게 도움이 될 수도.. 있기에..

#include <string>
#include <vector>

using namespace std;
vector<int> alphabetIndex(26,-1);

vector<int> solution(string s) {
    vector<int> answer;
    int sSize = s.size();
    for(int i =0; i < sSize; i++)
    {
        int c = s[i] - 'a';
        if(alphabetIndex[c] == -1)
        {
            answer.push_back(-1);
        }
        else
        {
            answer.push_back(i - alphabetIndex[c]);
        }
        alphabetIndex[c] = i;
    }
    return answer;
}