JuSeok

1806번

문제

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

문제 해결 순서

  1. 시작과 끝을 가르키는 s, e 변수를 선언한다.
  2. 끝을 늘려가며 합을 구해 S보다 큰 지 검사
  3. 크다면 (끝 - 시작) 값을 정답에 넣고, 시작 포인터를 +1 해주고, 합에서 현재 시작값을 빼줌.
  4. 끝이 범위를 벗어나거나 시작이 범위를 벗어나면 종료.
import java.util.Scanner;

public class Main {

    static int N, S, ans, sum;
    static int[] data;
    static int s, e;

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);

        N = sc.nextInt();
        S = sc.nextInt();

        data = new int[N + 1];
        for (int i = 0; i < N; i++) {
            data[i] = sc.nextInt();
        }

        ans = Integer.MAX_VALUE;
        s = 0;
        e = 0;
        while (e <= N && s <= N) {
            if (sum >= S && ans > e - s)
                ans = e - s;

            if (sum < S)
                sum += data[e++];
            else
                sum -= data[s++];
        }
        System.out.println(ans == Integer.MAX_VALUE ? 0 : ans);
    }
}

Tags