Programmers Level 1 수박수박수박수박수박수?

def water_melon(n):

    if n%2==0:

        return "수박"*(n//2)

    elif n==1: 

        return '수'*(n)

    elif n%2==1:

        return '수박'*(n//2)+'수'

# 실행을 위한 테스트코드입니다.

print("n이 9인 경우: " + water_melon(1));

print("n이 8인 경우: " + water_melon(8));