[Programmers] 문자열 나누기

문제

 

입출력 예시

 

개인적으로는 '입출력 예 #2'가 까다로웠다 🥲

 

풀이

처음에는 x=s[0]으로 미리 할당해놓고 s[1:]을 for문 돌리면서 문자 하나하나를 x와 비교하는 방식을 썼다. 이 방법으로 하니까 s 길이가 1일 때 인덱스 에러가 계속 떴는데, 인덱스 에러가 도저히 안 잡혀서 아예 생각을 다시 했다.

 

핵심은 '시작 문자를 정해주지 않는 것'이었다. for문 안에서 x, x_cnt, not_x_cnt 초기화하는 부분이 한 번 더 나오는 게 불편하지만 이건 더 줄일 수가 없는 부분이라서 어쩔 수 없었다.

 

def solution(s):
    answer = 0
    
    x_cnt = 0
    not_x_cnt = 0

    x = ''
    
    for alphabet in s:
        if x == '':
            x = alphabet
        
        if alphabet == x:
            x_cnt += 1
        else:
            not_x_cnt += 1
        
        if x_cnt == not_x_cnt:
            answer += 1
            
            x = ''
            x_cnt = 0
            not_x_cnt = 0
        
    if x:
        answer += 1
    
    return answer