[Programmers] 숫자의 표현

숫자의 표현

문제 설명

---

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까지의 총합을 나타내는 sums 리스트를 생성합니다.
2. l, r은 두 개의 포인터를 나타내며 투 포인터 알고리즘에 맞춰 진행하면서 원하는 값일 경우 cnt를 증가시킵니다.

def solution(n):
    sums = [0 for _ in range(n+1)] # n까지의 연속된 자연수의 합
    for i in range(1,n+1):
        sums[i] = sums[i-1] + i
        
    l,r,cnt = 0,1,0
    while l != len(sums)-1:
        mysum = sums[r] - sums[l]
        if mysum >= n:
            l += 1
            if mysum == n: cnt += 1
        else:
            if r == len(sums) - 1: break
            r += 1

    return cnt

위와 같은 코드를 실행시키면 된다.

그럼 안녕!