프로그래머스) 수박수박수박수박수박수? 문제 (파이썬)

문제 설명

 

길이가 n이고, "수박수박수박수...."와 같은 패턴을 유지하는 문자열을 리턴하는 함수, solution을 완성하세요.

 

예를들어 n이 4이면 "수박수박"을 리턴하고 3이라면 "수박수"를 리턴하면 됩니다.

 

제한 조건

• n은 길이 10,000이하인 자연수입니다.

 

입출력 예

 

3	"수박수"
4	"수박수박"

 

나의 풀이

 

def solution(a):
    if a % 2 == 0:
            return "수박"*(a//2)
    else:
            return "수박"*(a//2)+"수"

 

이번 문제는 홀수일 때 는 항상 뒤에 "수"가 붙는 것을 파악하여 a가 짝수(a%2==0)이면 수박*(a//2)를 반환하고,

 

홀수이면 짝수 뒤에 "수"를 더하였다.