프로그래머스 숫자의 표현 파이썬 - 2021/05/25

문제 설명

 

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

---

입출력 예

n                                                                                      result

15

4

입출력 예 설명

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

 

풀이

 

n의 범위가 10000이기 때문에 for문을 두번이용하면 시간초과가 날 수 있다. for문을 두 번이용한 풀이는 부분합을 구해놓고 부분합의 차이가 n이되는 경우를 count한다.

 

시간초과를 피하기 위해 투 포인터를 이용할 수 있다. for문 안에서 while문만 한번 돌기 때문에 시간초과를 피할 수 있다.

 

def solution(n):
    arr=[i for i in range(1,n+1)]
    hap=0
    end=0
    count=0
    for start in range(n):
        while hap<n and end<n:
            hap+=arr[end]
            end+=1
        if hap==n:
            count+=1
        hap-=arr[start]
        
        
    
    return count