[프로그래머스] Lv.0 - 잘라서 배열로 저장하기.py

문제

• 문자열 my_str과 n이 매개변수로 주어질 때, my_str을 길이 n씩 잘라서 저장한 배열을 return하도록 solution 함수를 완성해주세요.

제한사항

• 1 ≤ my_str의 길이 ≤ 100
• 1 ≤ n ≤ my_str의 길이
• my_str은 알파벳 소문자, 대문자, 숫자로 이루어져 있습니다.

입출력 예시

• 입출력 예 1

  • "abc1Addfggg4556b" 를 길이 6씩 잘라 배열에 저장한 ["abc1Ad", "dfggg4", "556b"]를 return해야 합니다.

• 입출력 예 2

  • "abcdef123" 를 길이 3씩 잘라 배열에 저장한 ["abc", "def", "123"]를 return해야 합니다.

유의사항

입출력 예 #1의 경우 "abc1Addfggg4556b"를 길이 6씩 자르면 "abc1Ad", "dfggg4" 두개와 마지막 "556b"가 남습니다. 이런 경우 남은 문자열을 그대로 배열에 저장합니다.

코드 １

def solution(my_str, n):
    answer = []
    temp = ''
    for ch in my_str:
        temp += ch
        if len(temp) == n:
            answer.append(temp)
            temp = ''
    if len(temp) >= 1:
        answer.append(temp)
    return answer

코드 ２

def solution(my_str, n):
    answer = []
    for itv in range(0, len(my_str), n):
        answer.append(my_str[itv:itv+n])
    return answer

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