알고리즘/기타

[프로그래머스 : PYTHON] 수박수박수박수박수박수?

cijbest 2021. 6. 5. 15:59

문제 설명

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

제한 조건

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

입출력 예

n return
3 "수박수"
4 "수박수박"

 

문제풀이

인덱스를 이용하여 '수', '박' 패턴을 만든다.

다른 사람 풀이는 '수박'을 n만큼 만들어서 n자리만큼 잘라서 리턴하는데 코드는 줄어들지만 불필요한 메모리가 발생할 수 있다.

 

전체코드

def solution(n):
    answer = ''
    
    for i in range(1, n+1):
        if i % 2:
            answer += '수'
        else:
            answer += '박'
            
    return answer

 

다른 사람 풀이

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

 

반응형