[test]수박수박수박수박수박수? - 파이썬

문제 설명
길이가 n이고, 수박수박수박수....와 같은 패턴을 유지하는 문자열을 리턴하는 함수, solution을 완성하세요. 예를들어 n이 4이면 수박수박을 리턴하고 3이라면 수박수를 리턴하면 됩니다.
제한 조건
n은 길이 10,000이하인 자연수입니다.
입출력 예
n return
3 수박수
4 수박수박

내 풀이

def solution(n):
    str = "수박"
    return str*(n//2)+"수" if n%2 == 1 else str*(n//2) 

1. 규칙 찾기
   "수박"은 n을 2로 나눴을 때 몫 만큼 반복되고 n이 홀수이면 뒤에 "수"가 붙음.
2. str에 반복되는 부분을 저장하고, 조건부 표현식으로 써줌.

다른 풀이

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

간단하게 "수박"을 충분히 써준다음 앞에서부터 n전까지 슬라이스