[python] 프로그래머스 - 수박수박수박수박수박수?

---

나의 풀이

def solution(n):
    if n%2 != 0: # 홀수
        return "수박"*(n//2)+"수"
    else: #짝수
        return "수박"* (n//2)

solution(5) # 수박수박수

 

홀수인 경우, 짝수인 경우 나누어했다.

 

---

모범 답안

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

print("n이 3인 경우: " + water_melon(3)); # 수박수
print("n이 4인 경우: " + water_melon(4)); # 수박수박수

 

파이썬의 강력한 기능인 인덱싱 기능을 잘 활용한 풀이인 것 같다!