[프로그래머스] 숫자의 표현 - python

숫자의 표현

문제

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 이하의 자연수 입니다.

풀이

슬라이딩 윈도우로 풀기

하나씩 더하다가 범위를 벗어나면 시작점을 옮겨준다.

def solution(n):
    answer = 0
    
    if n <= 1:
        return 1
    
    numbers = [i for i in range(1, n + 1)]
    i = 0
    tmp = numbers[i]
    
    for j in range(1, n):
        tmp += numbers[j]
        while tmp >= n and i < n:
            if tmp == n:
                answer += 1
            tmp -= numbers[i]
            i += 1
    return answer