[python] 프로그래머스 Lv 1 - 문자열 나누기

🔗 문제 링크

프로그래머스

📁 분류

구현

💡 풀이

주어진 규칙에 따라 문자열을 나누어서 나누어진 문자열의 개수를 반환하면 되는 문제이지만, 규칙을 이해하기가 매우 어려웠다.

주어진 규칙을 설명하자면 다음과 같다.

1. 문자를 하나씩 추가한다.
2. 추가되는 문자열에서 첫 문자의 개수와 나머지 개수를 구한다.
3. 첫 문자의 개수와 나머지 개수가 같다면 문자열을 초기화한다.
4. 1, 3 과정을 추가할 문자가 없을 때까지 반복한다.

💻 코드

def solution(s):
    answer = 0
    word = ''
    for i, ch in enumerate(s):
        word += ch
        count = word.count(word[0])
        other_count = len(word) - count

        if count == other_count:
            word = ''
            answer += 1

    return answer if len(word) == 0 else answer + 1