[알고리즘-JS, Python] 백준 2018번: 연속된 자연수의 합 (투 포인터)

어떠한 자연수 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을 입력받아 가지수를 출력하는 프로그램을 작성하시오.

 

풀이

JavaScript

const n = parseInt(require('fs').readFileSync("input.txt").toString().trim(), 10);

let count = 1; // 가장 마지막 수만 뽑는 경우의 수를 미리 넣고 시작
let start_index = 1; // 각 인덱스에 담긴 값이 1로 시작하므로 같이 하기 위함
let end_index = 1; // 각 인덱스에 담긴 값이 1로 시작하므로 같이 하기 위함
let sum = 1; // 각 인덱스에 담긴 값이 1로 시작하므로 초기 sum은 1

while (end_index !== n) {
	if (sum === n) {
    	count++; // 카운트 증가 먼저 한 후,
        end_index++; // end index 증가
        sum += end_index; // sum에 새로운 end_index 더함
    } else if (sum > n) {
    	sum -= start_index; // 현재 start_index를 먼저 빼주고,
        start_index++; // start_index 증가
    } else {
    	end_index++;
        sum += end_index;
    }
}

console.log(count);

 

 

Python

n = int(input())
start_index = 1
end_index = 1
sum = 1
count = 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)