지우너

[코드트리] 가장 짧은 부분합 C++ 본문

Problem Solving

[코드트리] 가장 짧은 부분합 C++

지옹 2024. 8. 22. 17:28

문제

https://www.codetree.ai/missions/8/problems/shortest-subtotal?&utm_source=clipboard&utm_medium=text

 

코드트리 | 코딩테스트 준비를 위한 알고리즘 정석

국가대표가 만든 코딩 공부의 가이드북 코딩 왕초보부터 꿈의 직장 코테 합격까지, 국가대표가 엄선한 커리큘럼으로 준비해보세요.

www.codetree.ai

 

코드

#include <iostream>
using namespace std;

const int MAX_N = 100001;

int n, s;
int num[MAX_N];

int main() {
    // input
    cin >> n >> s;
    for(int i=0; i<n; ++i){
        cin >> num[i];
    }
    
    int r=0, answer=2e9;
    int sum_val=0;
    for(int l=0; l<n; ++l){
        while(r+1<n && sum_val<s){
            sum_val+=num[r];
            r++;
        }
        if(sum_val>=s) answer = min(answer, r-l);
        sum_val-=num[l];
    }

    answer = (answer==2e9)? -1:answer;
    cout << answer << '\n';
    return 0;
}