[프로그래머스 / Python] 수박수박수박수박수박수?

https://programmers.co.kr/learn/courses/30/lessons/12922

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

 

길이가 1 일때,와 홀수일때 '수'만 잘 추가해주면 쉽게 풀수있다.

 

내 풀이

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

 

다른사람 풀이

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


def water_melon(n):
    return "수박"*(n//2) + "수"*(n%2)

이렇게 훨씬 간단하게 코드를 짤 수도있다.