코딩테스트

프로그래머스 140108 파이썬

피갓자 2025. 7. 3. 13:00

프로그래머스
프로그래머스

문자열 나누기

문제 설명

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

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

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

제한사항

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

입출력 예

s result
"banana" 3
"abracadabra" 6
"aaabbaccccabba" 3

입출력 예 설명

입출력 예 #1
s="banana"인 경우 ba - na - na와 같이 분해됩니다.

입출력 예 #2
s="abracadabra"인 경우 ab - ra - ca - da - br - a와 같이 분해됩니다.

입출력 예 #3
s="aaabbaccccabba"인 경우 aaabbacc - ccab - ba와 같이 분해됩니다.

문제 분석

의사 코드(pseudo code)
1 변수 answer 0으로 초기화
2 cnt1과 cnt2 0으로 초기화
3 x를 문자열 s의 첫 글자로 설정
4 문자열 s를 왼쪽부터 오른쪽까지 한 글자씩 반복
  a cnt1과 cnt2가 같다면 answer를 1 증가 기준 문자를 현재 글자로 갱신
  b 현재 글자가 기준 문자 x와 같다면 cnt1을 1 증가
  c 아니라면 cnt2를 1 증가
5 answer 반환

solution.py

def solution(s):
    answer = 0
    cnt1 = 0
    cnt2 = 0
    x = s[0]

    for i in s:
        if cnt1 == cnt2:
            answer += 1
            x = i
        if i == x:
            cnt1 += 1
        else:
            cnt2 += 1
    return answer

복잡도 분석

문자열 s를 한 글자씩 한 번씩만 순회하므로 시간 복잡도는 O(n)입니다.
변수 모두 상수 크기의 변수만 사용하므로 공간복잡도는 O(1)입니다.
따라서 최종 시간 복잡도와 공간 복잡도는 O(n)입니다.