문자열 나누기

문제 설명

문자열 s가 입력되었을 때 다음 규칙을 따라서 이 문자열을 여러 문자열로 분해하려고 합니다.

• 먼저 첫 글자를 읽습니다. 이 글자를 x라고 합시다.
• 이제 이 문자열을 왼쪽에서 오른쪽으로 읽어나가면서, x와 x가 아닌 다른 글자들이 나온 횟수를 각각 셉니다. 처음으로 두 횟수가 같아지는 순간 멈추고, 지금까지 읽은 문자열을 분리합니다.
• s에서 분리한 문자열을 빼고 남은 부분에 대해서 이 과정을 반복합니다. 남은 부분이 없다면 종료합니다.
• 만약 두 횟수가 다른 상태에서 더 이상 읽을 글자가 없다면, 역시 지금까지 읽은 문자열을 분리하고, 종료합니다.

문자열 s가 매개변수로 주어질 때, 위 과정과 같이 문자열들로 분해하고, 분해한 문자열의 개수를 return 하는 함수 solution을 완성하세요.

---

제한사항

• 1 ≤ s의 길이 ≤ 10,000
• s는 영어 소문자로만 이루어져 있습니다.

풀이 방법

1. 문자열의 첫 글자의 인덱스를 나타낼 idx와 x와 같은 글자가 나온 횟수 count1, 다른 글자가 나온 횟수 count2를 선언한다.

2. x와 s[i]를 비교해서 true면 count1++, false면 count2++해준다.

3. count1 == count2면 answer++해주고 idx에 count1 + count2를 더하고 count1과 count2를 0으로 초기화한다.

4. count1 != count2인 상태에서 더 이상 읽을 글자가 없다면 ++answer를 리턴한다.

#include <string>
#include <vector>

using namespace std;

int solution(string s) {
    int answer = 0;
    int count1 = 0;
    int count2 = 0;
    int idx = 0;
    char x = s[idx];
    
    for(int i = 0; i < s.length(); i++){
        
        if(x == s[i]) count1++;
        else count2++;

        if(count1 == count2) {
            answer++;
            idx += count1 + count2;
            count1 = 0;
            count2 = 0;
            x = s[idx];
        }
        if ((count1 != count2) && i == s.length() - 1) return ++answer;
    }
}