일기 대신 코드 슬쩍

[프로그래머스] Lv2. 숫자의 표현 본문

코딩테스트/프로그래머스(Python)

[프로그래머스] Lv2. 숫자의 표현

코코자 2023. 2. 11. 20:59

문제 설명

Finn은 요즘 수학공부에 빠져 있습니다. 수학 공부를 하던 Finn은 자연수 n을 연속한 자연수들로 표현 하는 방법이 여러개라는 사실을 알게 되었습니다. 예를들어 15는 다음과 같이 4가지로 표현 할 수 있습니다.

  • 1 + 2 + 3 + 4 + 5 = 15
  • 4 + 5 + 6 = 15
  • 7 + 8 = 15
  • 15 = 15

자연수 n이 매개변수로 주어질 때, 연속된 자연수들로 n을 표현하는 방법의 수를 return하는 solution를 완성해주세요.

제한사항

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

입출력 예

nresult
154

입출력 예 설명

입출력 예#1문제의 예시와 같습니다.

문제 풀이(+5)

1트 > 효율성 테스트 실패

이중 for문을 제거해야 할 것 같습니다.

def solution(n):
    answer = 0
    for i in range(1,n+1):
        sum = 0
        for j in range(i,n+1):
            sum += j
            if sum == n:
                answer += 1
                break
    return answer

2트 > 이중 for문의 문제가 아니라, sum이 n을 넘어가면 break을 걸어주는 조건만 추가하면 되는 거 였습니다.(+5)

def solution(n):
    answer = 0
    for i in range(1,n+1):
        sum = 0
        for j in range(i,n+1):
            sum += j
            if sum == n: # 합계가 n이 되면 answer +1
                answer += 1
                break
            elif sum > n: # 합계가 n을 넘어가면 break
                break
    return answer