문자열 나누기 (python)

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

프로그래머스

문제 설명

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

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

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

 

제한사항

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

---

알고리즘

first_cha = s[0]
start = 0
first_cha_count = 1
other_cha_count = 0
answer = []

필요한 변수를 만들어 초기화한다.

first_cha: 나눌 문자열의 가장 앞 글자

start: 나눌 문자열의 가장 앞 글자의 인덱스

first_cha_count: 문자열에서 가장 앞글자와 같은 문자의 개수

other_cha_count: 문자열에서 가장 앞글자와 다른 문자의 개수

answer: 나눈 문자열을 저장할 배열

 

    for i in range(1, len(s)):
        if first_cha_count == other_cha_count:
            answer.append(s[start:i]) # 문자열 추가하기
            first_cha = s[i] # 다음 문자부터 인덱싱하기
            first_cha_count = 0
            other_cha_count = 0
            start = i
        if s[i] == first_cha:
            first_cha_count += 1
        else:
            other_cha_count += 1

문자열의 글자만큼 반복문으로 다음을 반복한다. 만약 문자열의 가장 앞글자와 같은 문자의 개수가 아닌 문자의 개수가 같으면 이전(현재 인덱스는 개수를 세아린 문자의 다음 인덱스이니까)까지의 인덱스까지 인덱싱을 통해 문자열을 answer에 추가한다. 문자열을 나눴으면 다시 문자열을 나눠야 하고, 새 문자열의 가장 앞글자와 그의 인덱스도 새로 지정해야 한다. 또한 이전까지 센 글자들의 개수를 초기화한다.

 

    if first_cha_count != 0:
        answer.append(s[start:])
    return len(answer)

반복문을 모두 돌고 나면 마지막으로 글자가 남았는데, answer에 포함되지 않은 경우도 있으니, 위 코드를 통해서 마지막 글자까지 문자열로 answer에 포함한다. 그리고 문자열의 개수를 답으로 반환한다.

 

# LV.1
# 문자열 나누기
def solution(s):
    first_cha = s[0]
    start = 0
    first_cha_count = 1
    other_cha_count = 0
    answer = []
    for i in range(1, len(s)):
        if first_cha_count == other_cha_count:
            answer.append(s[start:i]) # 문자열 추가하기
            first_cha = s[i] # 다음 문자부터 인덱싱하기
            first_cha_count = 0
            other_cha_count = 0
            start = i
        if s[i] == first_cha:
            first_cha_count += 1
        else:
            other_cha_count += 1
    if first_cha_count != 0:
        answer.append(s[start:])
    return len(answer)