[Python] 프로그래머스 문자열 나누기

문제 설명

문자열 s가 입력되었을 때 다음 규칙을 따라서 이 문자열을 여러 문자열로 분해하려고 합니다.
먼저 첫 글자를 읽습니다. 이 글자를 x라고 합시다.
이제 이 문자열을 왼쪽에서 오른쪽으로 읽어나가면서, x와 x가 아닌 다른 글자들이 나온 횟수를 각각 셉니다. 처음으로 두 횟수가 같아지는 순간 멈추고, 지금까지 읽은 문자열을 분리합니다.
s에서 분리한 문자열을 빼고 남은 부분에 대해서 이 과정을 반복합니다. 남은 부분이 없다면 종료합니다.
만약 두 횟수가 다른 상태에서 더 이상 읽을 글자가 없다면, 역시 지금까지 읽은 문자열을 분리하고, 종료합니다.
문자열 s가 매개변수로 주어질 때, 위 과정과 같이 문자열들로 분해하고, 분해한 문자열의 개수를 return 하는 함수 solution을 완성하세요.

 

https://school.programmers.co.kr/learn/courses/30/lessons/140108

프로그래머스

제한 사항

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

예제

s	result
banana	3
abracadabra	6
aaabbaccccabba	3

문제 풀이

def solution(s):
    answer = 0
    idx = 0

    while idx < len(s):
        x = s[idx]  # 기준 문자
        x_cnt = 1  # 기준 문자의 개수 (기준 문자 포함)
        no_cnt = 0  # 기준 문자가 아닌 문자의 개수
        
        for i in range(idx + 1, len(s)):
            if s[i] == x:
                x_cnt += 1
            else:
                no_cnt += 1

            # 기준 문자와 다른 문자의 개수가 같아지면 분할
            if x_cnt == no_cnt:
                idx = i  # 현재 위치까지 분할
                break
        else:
            # 기준 문자와 다른 문자의 개수가 같아지지 않는 경우
            idx = len(s) - 1
        
        answer += 1  # 분할된 문자열의 개수 증가
        idx += 1  # 다음 위치로 이동
    
    return answer

---

파이썬을 독학하시는 분들에게 도움이 되길 바라며,
혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.