본문 바로가기

개인 공부/알고리즘

[Programmers] Level1 - 수박수박수박수박수박수?(연습문제) Python 풀이

728x90
반응형

문제

길이가 n이고, 수박수박수박수....와 같은 패턴을 유지하는 문자열을 리턴하는 함수, solution을 완성하세요. 예를들어 n이 4이면 수박수박을 리턴하고 3이라면 수박수를 리턴하면 됩니다.

입출력 예

n return
3 수박수
4 수박수박

풀이

이 문제는 예시로 알려준 입출력을 보게 되면 힌트를 쉽게 얻을 수 있다.
n은 자연수이며, 반환되는 값은 n의 값에 따라 '수' / '박'으로 끝이 나게 된다.
즉 n이 홀수인 경우 '수'로 끝이 나고 n이 짝수인 경우 '박'으로 끝이 나게 됨을 쉽게 알 수 있다.
따라서 n번 반복하여 홀수/짝수를 구분하여 값을 반환하면 해결이 된다.

코드

def solution(n):
    answer = ''

    for i in range(n):
        if i % 2 == 0:
            answer += '수'
        else:
            answer += '박'

    return answer
728x90
반응형