[프래그래머스] 수박수박수박수?(Python)

첫 코드

def solution(n):
    if n % 2 == 0:
        answer = (n//2 * '수박')

    elif n % 2 == 1:
        answer = ((n//2 * '수박')+'수')
    return answer

개선 코드

def solution(n):
    answer = '수박' * (n//2)
    if n % 2 == 1:
        return answer + '수'
    
    return answer

위의 코드처럼 간결하게 개선할 수 있다.