[백준/C++] 1806번: 부분합

풀이

#include <iostream>
#include <algorithm>

using namespace std;


int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    int N, S;
    cin >> N >> S;

    int A[N];
    for( int i = 0; i < N; ++i ) cin >> A[i];
    // 수열 입력

    int sum;
    int lft = 0, rgt = 0;
    sum = A[lft];

    int result = 100000001;

    while(rgt < N) {
        if( sum < S ) {
            rgt++;
            sum += A[rgt];
            continue;
        } else if( sum > S ) {
            result = min(result, (rgt - lft + 1));
            sum -= A[lft];
            lft++;
            continue;
        }
        result = min(result, (rgt - lft + 1));
        rgt++;
        sum += A[rgt];
    }

    if( result == 100000001 ) cout << 0; else cout << result;

    return 0;
}

1) lft, rgt 두 포인터 변수를 선언, 처음 부분합 sum은 A[0]로 설정

2) while문(rgt가 N보다 작은 동안)에서 최소 길이를 result 변수로 설정
- 1 ) sum이 S보다 작다면 rgt의 값을 증가시키고, 부분합에 추가
- 2 ) sum이 S보다 크다면 result의 값을 업데이트 해주고, 부분합에서 lft위치에 있는 원소를 빼주고 lft 값을 증가
- 3 ) 이외에는 sum과 S가 같은 경우이므로, 조건문의 마지막과 같이 식 작성
(지금 글을 쓰면서 보니, 굳이 S와 값이 같은 경우는 포함하지 않아도 괜찮을 것 같다는 생각이.....)

3) result의 값을 불가능한 큰 수로 설정해놨는데 갱신이 안되었다면 가능한 경우가 존재하지 않으므로 0 출력, 아니라면 result를 출력해 완성