[백준 P1806] 부분합 Java

P1806 부분합

1806번: 부분합

문제

10,000 이하의 자연수로 이루어진 길이 N짜리 수열이 주어진다. 이 수열에서 연속된 수들의 부분합 중에 그 합이 S 이상이 되는 것 중, 가장 짧은 것의 길이를 구하는 프로그램을 작성하시오.

입력값

첫째 줄에 N (10 ≤ N < 100,000)과 S (0 < S ≤ 100,000,000)가 주어진다. 둘째 줄에는 수열이 주어진다. 수열의 각 원소는 공백으로 구분되어져 있으며, 10,000이하의 자연수이다.

10 15
5 1 3 5 10 7 4 9 2 8

출력값

첫째 줄에 구하고자 하는 최소의 길이를 출력한다. 만일 그러한 합을 만드는 것이 불가능하다면 0을 출력하면 된다.

2

알고리즘

S 이상이 되는 것 중에 가장 짧은 것을 구하면 된다. 부분합이 존재한다고 가정했을 때 하나의 원소가 S 보다 크다면 정답은 1이 된다.

가장 처음에는 start와 end 변수에 0을 할당하여 배열의 가장 앞 부분을 가리키게 한다. 현재 값을 저장하기 위해서 current 변수에 배열의 가장 앞 부분의 값을 할당한다.

만약 그 값이 S 보다 작다면 더 많은 수를 더해서 current 의 값을 높여야 하므로 end 포인터를 1 증가 시킨다.

만약 그 값이 S 이상이라면 우선 정답 조건에 만족하는 값을 찾았으므로 정답을 업데이트한다. 이보다 짧은 부분합이 존재할 가능성이 있으므로 start 포인터를 1 증가한다.

start가 end 보다 커지는 순간에 종료한다.

코드

import java.util.*;
import java.io.*;

public class Main {
    static int N;
    static long S;
    static long[] numList;
    static long answer;

    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        N = Integer.parseInt(st.nextToken());
        S = Integer.parseInt(st.nextToken());
        numList = new long[N + 1];
        answer = Long.MAX_VALUE;

        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < N; i++) {
            numList[i] = Long.parseLong(st.nextToken());
        }

        int start = 0, end = 0;
        long current = numList[end];

        while (start < N && end < N && start <= end) {
            if (current < S) {
                current += numList[++end];
            } else {
                answer = Math.min(answer, end - start + 1);
                current -= numList[start++];
            }
        }
        System.out.println(answer == Long.MAX_VALUE ? 0 : answer);
    }
}