숫자의 표현 [Level 2] Python

https://data-make.tistory.com/307 -> 원리

문제 설명

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

입출력 예

입출력 예 설명

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

이미지 출처: https://bhsmath.tistory.com/17

코드

# https://data-make.tistory.com/307 -> 원리

def solution(n):
    answer = 0
    
    for i in range(1, n//2+2): # n//2의 범위로 했을 때 실패 되는 케이스들도 있어서 +2 만큼의 범위를 확대함
        num = n - i*(2+(i-1))//2 #  n - 등차수열의 합
        if num >= 0:
            if num % i == 0:
                #print([x+num//i for x in range(1, i+1)]) # 연속된 자연수들 출력
                answer += 1

    return answer # 테스트 케이스 15, 18은 range 범위를 +1을 해줘야 됨