Programmers / 수박수박수박수박수박수? / Python

문제 설명

길이가 n이고, "수박수박수박수...."와 같은 패턴을 유지하는 문자열을 리턴하는 함수, solution을 완성하세요. 예를들어 n이 4이면 "수박수박"을 리턴하고 3이라면 "수박수"를 리턴하면 됩니다.

제한 조건

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

입출력 예nreturn

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

---

내가 한 풀이 

기본적으로 몫과 나머지를 이용해서 n이 짝수, 홀수인 경우로 생각해서 답안 작성

def solution(n):
    answer = '수박'
    
    if n == 1:
        return str(answer[0])
    elif n % 2 == 0:
        return str(answer * (n//2))
    return str(answer * (n//2) + answer[0])

 

모범답안

 

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