[프로그래머스] Lv.1 문자열 나누기 (파이썬)

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

프로그래머스

 

문제 설명

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

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

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

 

제한사항

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

---

나의 풀이

def solution(s):
    answer = 0
    while s:
        x = s[0]
        x_count, other = 0, 0
        while x_count == other:
            for i in s:
                if i == x:
                    x_count += 1
                else:
                    other += 1
        answer += 1
        s = s[x_count*2:]
    return answer

 

• while을 써서 반복 루프를 돌리려고 했는데... 무한 루프에 빠질 가능성 발견 ! (어쩐지 시간초과 뜨더라) (그냥 for문 쓸 걸,, 사람이 안 하던 짓을 하면 안됨 ㅠㅠ)
• s 를 다시 지정할 때 뭔가 이상하게 지정됨
• 걍 문제점이 많음.

 

GPT가 디버깅 해준 풀이

def solution(s):
    answer = 0
    while s:
        x = s[0]
        x_count, other = 0, 0
        for i, char in enumerate(s):
            if char == x:
                x_count += 1
            else:
                other += 1
            
            if x_count == other:
                answer += 1
                s = s[i+1:]
                break
        else:
            answer += 1
            s = ''
                
    return answer

 

• 이상한게 꼭 enumerate을 쓰고 싶을 땐 안쓰는게 낫고, 안써야겠다 다짐하고 풀면 쓰는 게 맞더라,,,?
• 아무튼! x_count, other을 각각 계산해주고, 뒤에 if문으로 같은지 확인.
• s 슬라이싱 한다음에 break 하고 다시 처음부터 시작 ~
• s 가 비어있거나 1개 남아있거나 등등일 때는 그냥 answer += 1 해주고 s 빈 문자열 만들어주면 while에서 빠져나옴