(Python) 프로그래머스 : 수박수박수박수박수박수?

def solution(n):
    answer = ''
    num = n//2
    str1 = "수박"
    if n%2 == 0:
        answer = str1 * num
    else:
        answer = (str1 * num) + "수"
    return answer