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

[문제 설명]

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

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

 

[제한사항]

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

 

[입출력 예]

[문제 풀이]

def solution(s):
    answer = 0
    fwc=dc=0
    for i in range(len(s)):
        if i==0:
            fw=s[0]
            fwc+=1
        else:
            if s[i]==fw:
                fwc+=1
            else:
                dc+=1 
        if fwc==dc:
            try:
                answer+=1
                fwc=dc=0
                fw=s[i+1]
            except:
                pass
        
        if i==len(s)-1:
            if fwc!=0 or dc!=0:
                answer+=1
            
    return answer

첫글자 갯수 fwc, 다른 문자 갯수 dc로 두고 해결.

마지막 if문은 fwc, dc가 같게 끝나지 않았을 경우 대비한 코드.

def solution(s):
    answer = 0
    sav1=0
    sav2=0
    for i in s:
        if sav1==sav2:
            answer+=1
            a=i
        if i==a:
            sav1+=1
        else:
            sav2+=1
    return answer

위의 코드는 다른 분들의 풀이 과정.

 

sav1, sav2를 처음에 비교하면 0일 경우 더해지기 때문에 위 방식을 사용하지 않았음.

하지만 위의 방식대로 코드를 짜면 sav1과 sav2가 같게 끝나지 않았을 경우의 코드를 짜지 않아도 돼서 더 간편.

 

출처 : 프로그래머스 https://school.programmers.co.kr/learn/courses/30/lessons/140108