[알고리즘/파이썬] 자료구조 - 투 포인터(Two pointers) | 백준 2018번: 수들의 합 5

 

 

 

투 포인터(Two Pointers) 알고리즘

- 정렬된 배열의 부분합 또는 개수 구하기 등의 문제를 해결하는 데 자주 사용되는 알고리즘이다. 

- 기본 아이디어는 배열의 시작과 끝에서 하나씩 시작해서, 조건에 따라 양쪽 방향으로 포인터를 이동해 문제해결한다.

- 문제에서 배열에 대한 순서나 시작점이 중요하지 않다면 이 알고리즘을 사용해서 빠르게 문제를 해결할 수 있다.

 

 

 

 

---

 

 

 

 

 

2018번: 수들의 합 5

문제

어떠한 자연수 N은, 몇 개의 연속된 자연수의 합으로 나타낼 수 있다. 당신은 어떤 자연수 N(1 ≤ N ≤ 10,000,000)에 대해서, 이 N을 몇 개의 연속된 자연수의 합으로 나타내는 가지수를 알고 싶어한다. 이때, 사용하는 자연수는 N이하여야 한다.

예를 들어, 15를 나타내는 방법은 15, 7+8, 4+5+6, 1+2+3+4+5의 4가지가 있다. 반면에 10을 나타내는 방법은 10, 1+2+3+4의 2가지가 있다.

N을 입력받아 가지수를 출력하는 프로그램을 작성하시오.

입력

첫 줄에 정수 N이 주어진다.

출력

입력된 자연수 N을 몇 개의 연속된 자연수의 합으로 나타내는 가지수를 출력하시오

 

n = int(input())
count = 1
start_index = 1
end_index = 1
_sum = 1

while end_index != n:
    if _sum == n: 
        count += 1
        end_index += 1
        _sum += end_index
    elif _sum > n:
        _sum -= start_index
        start_index += 1
    else:
        end_index += 1
        _sum += end_index
print(count)

 

반응형

현재 합이 n과 같으면

 if _sum == n: 
        count += 1
        end_index += 1
        _sum += end_index

1. 경우의 수를 1 증가시킨다

2. 끝점을 이동시킨다.

3. 새로운 끝점을 합에 더한다.

 

 

현재 합이 n보다 크면

elif _sum > n:
        _sum -= start_index
        start_index += 1

1. 시작점을 빼면서 합을 줄인다.

2. 시작점을 이동시킨다.

 

 

현재 합이 n보다 작으면

else:
        end_index += 1
        _sum += end_index

1. 끝점을 이동시킨다.

2. 새로운 끝점을 합에 더한다.