[BOJ] 1806. 부분합

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

1806번: 부분합

수열에서 연속된 수들의 부분합 중에 그 합이 S 이상이 되는 것 중, 가장 짧은 것의 길이를 구하는 문제이다.

 

left와 right 변수 두 개를 포인터로 놓고 구해줬다.

left는 부분 수열의 시작 부분, right는 부분 수열의 마지막 부분을 가리킨다.

 

가장 짧은 것의 길이를 구하기 위해 최대 길이인 n+1을 ans에 저장해놓는다.

그럼 이제 부분합을 s와 비교해가며 구해주면 된다.

 

먼저 합이 s보다 작은 경우에는 right를 증가시켜서 합에 수열의 원소를 더해준다.

합이 s보다 클 경우에는 left가 가리키는 원소를 합에서 빼준 후에 left값을 뒤로 옮긴다.

이때, s보다 큰 경우도 정답에 포함이므로 그때의 부분 수열의 길이(right-left+1)를 구해준다.

마찬가지로 합이 s가 되었을 때도 부분 수열의 길이를 구해주면 된다.

 

이 과정을 left가 right보다 커져버리거나 right가 범위를 벗어나는 경우가 될 때까지 반복한다.

 

합이 s이상이 되는 경우가 없다면 처음에 저장해 놓은 n+1 값이 ans에 저장되어 있으므로

ans가 그대로 n+1 인 경우에는 문제 조건에 따라 0을 출력한다.

 

#include <iostream>
using namespace std;
 
int n, s;
int num[100000];
 
int main()
{
 
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
 
    cin >> n >> s;
 
    for (int i = 0; i < n; i++) {
        cin >> num[i];
    }
    
    int ans = n+1;
    
    int sum = num[0];
    int left = 0;
    int right = 0;
 
    while(left <= right && right < n) {
        
        if (sum < s) {
            right++;
            sum += num[right];
        }
        else if (sum > s) {
            int len = right - left + 1;
            if (ans > len) ans = len;
 
 
            sum -= num[left];
            left++;
        }
        else {
            int len = right - left + 1;
            if (ans > len) ans = len;
 
 
            right++;
            sum += num[right];
        }
 
    }
 
    if (ans == n + 1) ans = 0;
    cout << ans << '\n';
 
 
    return 0;
}