프로그래머스 lv1 수박수박수박수박수박수(연습문제)

문제 설명

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

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

제한 조건

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

입출력 예

접근

짝수 일때 "박"으로 끝나고 홀 수 일때 "수"로 끝난다는 점에 주목하면 쉽게 풀 수 있음

풀이

# 내 풀이
def solution(n):
    answer = ''
    for i in list(range(1, n+1)):
        if i % 2 == 1:
            answer += "수"
        else:
            answer += "박"
    
    return answer
    
if __name__ == "__main__":
    n = 3#return "수박수"
#     n = 4 #return "수박수박"
    print(solution(n))

"수박"을 보면 인덱스가 홀수 일때는 "수" 짝수 일 때는"박"이므로 전체 길이에 맞게 "수"와 "박"을 더해주면 됨. 쉬운 문제

# 다른 사람 풀이

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

//로 나누면 소수점 제외 정수로 반환해줌. n의 짝수번째만큼 "수박"을 더하고, %는 나머지를 반환하므로 2로 나눴을때 나머지는 홀수다. 나머지가 있다면 n은 홀수 이므로 마지막에 "수"를 더함.

출처: 프로그래머스 코딩 테스트 연습, https://programmers.co.kr/learn/challenges