반응형

문제 링크: https://www.acmicpc.net/problem/1806

 

1806번: 부분합

문제 10,000 이하의 자연수로 이루어진 길이 N짜리 수열이 주어진다. 이 수열에서 연속된 수들의 부분합 중에 그 합이 S 이상이 되는 것 중, 가장 짧은 것의 길이를 구하는 프로그램을 작성하시오. 입력 첫째 줄에 N (10 ≤ N < 100,000)과 S (0 < S ≤ 100,000,000)가 주어진다. 둘째 줄에는 수열이 주어진다. 수열의 각 원소는 공백으로 구분되어져 있으며, 10,000이하의 자연수이다. 출력 첫째 줄에 구하고자 하는 최소의 길

www.acmicpc.net

<문제 풀이> 투 포인터

투 포인터 알고리즘을 적용하면 된다.

 

left, right 포인터를 준비하고 찾은 부분합이 s이상이면 그때의 수열의 길이와 이전에 구했던 길이의 최솟값으로 갱신하고 left을 증가시키면 된다. 왜냐하면 부분합이 s이상인걸 찾았을 때 right을 증가시키면 수열의 길이가 증가하므로 확인할 필요가 없다.

 

<C++ 소스 코드>

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
#include <iostream>
#include <algorithm>
using namespace std;
 
int n, s;
int arr[100000];
int main(void) {
    ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
    cin >> n >> s;
    for (int i = 0; i < n; i++cin >> arr[i];
    int l = 0, r = 0, sum = 0, res = n;
    bool existence = false;
    while (true) {
        if (sum >= s && s) {
            existence = true;
            res = min(res, r - l);
            sum -= arr[l++];
        }
        else if (r == n) break;
        else sum += arr[r++];
    }
    if (existence) cout << res;
    else cout << 0;
    return 0;
}
cs

 

 

반응형

+ Recent posts