CODE/Algorithm (Python)

프로그래머스 level 1 문자열 나누기

sed 2026. 4. 27. 16:49
SMALL
문자열 나누기
제출 내역
 
문제 설명

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

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

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


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

입출력 예sresult
"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와 같이 분해됩니다.

 

아래와 같이 코드를 작성했다.

char에 비교될 문자(x)를 넣고, for 문을 돌면서 x와 같다면 += 1 하고 그렇지 않다면 not_x에 1을 추가한다.

만일 x와 not_x가 같다면 분할된 문자열 체크 수인 cnt를 +1 한다. 동시에 문자열 s를 잘라서 이미 체크했던 부분(분할된 부분을 제거한다)을 없애 새로운 문자열 s로 업데이트한다.

def solution(s):
    cnt = 0
    
    while s:
        char = s[0]
        x = 1
        not_x = 0

        for i in range(1, len(s)):
            if s[i] == char:
                x += 1
            else:
                not_x += 1
            
            if x == not_x:
                cnt += 1
                s = s[i+1:]
                break
        else:
            cnt += 1
            break
                
    return cnt

 

 

그런데 인덱스로 수행하면 더 효율적으로 동작할 수 있다는 것을 알게 되었다.

 

문자열 s의 길이를 n으로 두고, 비교할 문자 x 보다 n이 작을 때 while 문을 돌게한다.

첫번째 문자는 문자열 s의 start를 반환하게 하여, 다음 문자열(i = start + 1)이 첫번째 문자와 같을 때 x를 +1 한다. 그렇지 않으면 not_x를 +1 한다. 

x와 not_x가 같을 때는 break를 걸어 준다.

분할된 문자열의 체크 수인 answer 을 +1 해준 뒤, 다음 문자열 인덱스로 넘어간다(start = i+1).

def solution(s):
    answer = 0
    start = 0
    n = len(s)

    while start < n:
        first = s[start]
        x = 1
        not_x = 0

        i = start + 1
        while i < n:
            if s[i] == first:
                x += 1
            else:
                not_x += 1

            if x == not_x:
                break
            i += 1

        answer += 1
        start = i + 1  # 다음 구간으로 점프

    return answer
LIST