[백준] 2018 수들의 합 5 (투 포인터) - Java

 

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의 최대값이 10,000,000으로 O(nlogn)의 시간복잡도 알고리즘을 사용하면 시간초과가 난다.
• O(n)의 시간복잡도 알고리즘을 사용해야 하기 때문에 투포인터를 이용한다.
• startIdx 와 endIdx를 지정하여 조건에 맞게 인덱스값을 더해가며 문제를 푼다.

 

문제 풀이

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int N = sc.nextInt();

        int count = 1;
        int sum = 1;

        int startIdx = 1;
        int endIdx = 1;

        while( endIdx != N) { 

            if(sum < N){
                endIdx++;
                sum += endIdx;
            } else if(sum== N) {
                count++;
                endIdx++;
                sum += endIdx;
            }else {
                sum -= startIdx;
                startIdx++;
            }

        }

        System.out.println(count);
    }
}