1806번 : 부분합 [Java]

2021. 4. 20. 21:53Algorithm/백준

반응형

https://www.acmicpc.net/problem/1806

 

1806번: 부분합

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

www.acmicpc.net

 

해결방안 

모든 값을 배열에 넣은 뒤 투포인터를 사용해 S이상이 되는 경우를 찾고, 그중 최솟값을 기억한다.

 

주의할 점

처음에 투포인터를 사용하지 않고 for문을 두 개 사용했더니 (for (int i = 0; i < N; i++) /  for (int j = i; j < N; j++)) 시간 초과가 났다. 제발 범위 확인 좀 하자,,

 

전체 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {
	public static void main(String[] args) throws IOException {
		BufferedReader input = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(input.readLine(), " ");
		int N = Integer.parseInt(st.nextToken());
		int S = Integer.parseInt(st.nextToken());
		int[] arr = new int[N];
		st = new StringTokenizer(input.readLine(), " ");
		for (int i = 0; i < N; i++)
			arr[i] = Integer.parseInt(st.nextToken());

		int ans = 987654321;
		int start = 0;
		int end = 0;
		int sum = 0;

		while (true) {

			if (sum >= S) {
				sum -= arr[start];
				start++;
				ans = Math.min(ans, end - start + 1);
			} else if (end == N) // 끝까지 다 봤다.
				break;
			else {
				sum += arr[end];
				end++;
			}
		}
		System.out.println(ans != 987654321 ? ans : 0);
	}
}
반응형

'Algorithm > 백준' 카테고리의 다른 글

17471 번 : 게리맨더링 [Java]  (0) 2021.04.23
16973번 : 직사각형 탈출 [Java]  (0) 2021.04.20
1644번 : 소수의 연속합 [Java]  (0) 2021.04.20
13305번 : 주유소 [Java]  (0) 2021.04.12
1987번 : 알파벳 [Java]  (0) 2021.04.11