[프로그래머스] Lv.0 - x 사이의 개수.py

문제

• 문자열 myString이 주어집니다. myString을 문자 "x"를 기준으로 나눴을 때 나눠진 문자열 각각의 길이를 순서대로 저장한 배열을 return 하는 solution 함수를 완성해 주세요.

제한사항

• 1 ≤ myString의 길이 ≤ 100,000
• myString은 알파벳 소문자로 이루어진 문자열입니다.

입출력 예시

• 입출력 예 1
  • "x"를 기준으로 문자열을 나누면 ["o", "oo", "o", "", "o", ""]가 됩니다. 각각의 길이로 배열을 만들면 [1, 2, 1, 0, 1, 0]입니다. 따라서 [1, 2, 1, 0, 1, 0]을 return 합니다.
• 입출력 예 2
  • "x"를 기준으로 문자열을 나누면 ["", "abc", "def", "ghi"]가 됩니다. 각각의 길이로 배열을 만들면 [0, 3, 3, 3]입니다. 따라서 [0, 3, 3, 3]을 return 합니다.

코드 1

def solution(myString):
    answer = []
    count = 0
    for val in myString:
        if val == 'x':
            answer.append(count)
            count = 0
        else:
            count += 1
    answer.append(count)
    return answer

코드 2

def solution(myString):
    answer = []
    myString = myString.split('x')
    for val in myString:
        answer.append(len(val))
    return answer

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges