Programmers - 숫자의 표현[파이썬(python)]

문제 출처 : https://programmers.co.kr/learn/courses/30/lessons/12924

코딩테스트 연습 - 숫자의 표현

문제 설명

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
문제의 예시와 같습니다.

---

문제 풀이

투 포인터를 활용하여 문제를 풀었다. 이 문제의 핵심은 수의 합이 연속된 수여야 한다는 것이다.

이럴 경우 1부터 n까지의 수를 나열하고 연속된 구간의 합이 n이 되면 count를 하는 방식으로 접근할 수 있고,

구간을 선정하기 위해서 start 포인터와 end 포인터가 필요하다.

알고리즘은 다음과 같다.

1. 1부터 n까지 수를 저장한 배열을 생성한다.
2. 배열의 index를 의미하는 start와 end를 0으로 초기화한다.
3. 배열의 start부터 end까지의 수의 합이 n과 같다면 end와 start포인터를 1 증가시킨다.
4. 배열의 start부터 end까지의 합이 n보다 작다면 합이 커지게 하기 위해 end 포인터를 1 증가시킨다.
5. 배열의 start부터 end까지의 합이 n보다 크다면 합이 작아지게 하기 위해 start 포인터를 1 증가시킨다.

 

구현 코드

코드 링크 : 숫자의 표현

def solution(n):
    answer = 0
    nums = [ i for i in range(1, n+1)]
    start = 0
    end = 0
    while start < n:
        if sum(nums[start:end]) == n:
            end += 1
            start += 1
            answer += 1
        elif sum(nums[start:end]) < n:
            end += 1
        else:
            start += 1
    return answer