Data Structure & Algorithm
[알고리즘-JS, Python] 백준 2018번: 연속된 자연수의 합 (투 포인터)
fnow
2024. 3. 17. 20:29
반응형
어떠한 자연수 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)
반응형