[프로그래머스] Lv.0 - 공백으로 구분하기 2.py

문제

• 단어가 공백 한 개 이상으로 구분되어 있는 문자열 my_string이 매개변수로 주어질 때, my_string에 나온 단어를 앞에서부터 순서대로 담은 문자열 배열을 return 하는 solution 함수를 작성해 주세요.

제한사항

• my_string은 영소문자와 공백으로만 이루어져 있습니다.
• 1 ≤ my_string의 길이 ≤ 1,000
• my_string의 맨 앞과 맨 뒤에도 공백이 있을 수 있습니다.
• my_string에는 단어가 하나 이상 존재합니다.

입출력 예시

• 입출력 예 1
  • 예제 1번의 my_string은 " i love you"로 공백을 기준으로 단어를 나누면 "i", "love", "you" 3개의 단어가 있습니다. 따라서 ["i", "love", "you"]를 return 합니다.
• 입출력 예 2
  • 예제 2번의 my_string은 " programmers "로 단어는 "programmers" 하나만 있습니다. 따라서 ["programmers"]를 return 합니다.

코드 1

def solution(my_string):
    answer = []
    word = ''

    for idx, val in enumerate(my_string):
        if val == ' ':
            if len(word) >= 1:
                answer.append(word)
                word = ''
            else:
                continue
        else:
            word += val
            if (idx + 1) == len(my_string):
                answer.append(word)
    return answer

코드 2

def solution(my_string):
    answer = my_string.split()
    return answer

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