[프로그래머스][python]12922. 수박수박수박수박수박수?

MLra 2021. 1. 9. 15:01

코딩테스트 연습 - 수박수박수박수박수박수?

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

programmers.co.kr

My answer:

def solution(n):
    repeat=int(n/2)
    remain=n%2
    answer = ''
    answer+=repeat*("수박")
    if remain!=0:
        answer+="수"
    return answer

reference code:

def water_melon(n):
    s = "수박" * n
    return s[:n]