algorithm

[프로그래머스 LV.2 숫자의 표현]

선호하는 2023. 11. 7. 09:00

문제 링크:https://school.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 이하의 자연수 입니다.

<문제 풀이>
투 포인터 유형의 알고리즘 문제이다. [s:e]의 합이 기준보다 크면 시작지점을 이동하고, 기준보다 작으면 끝 지점을 이동하는 식으로 진행하면서 같은 경우를 찾으면 된다.

<소스코드>

def solution(n):
    answer = 0
    s, e = 1,1
    tmp = 1


    while s <= n:
        if tmp < n:
            e += 1
            tmp += e
        elif tmp > n:
            tmp -= s
            s += 1
        else:
            tmp -= s
            s += 1
            answer += 1

    return answer