245912026-02-12 21:27:13999Zenehallgatáscpp17Accepted 50/5056ms2572 KiB
// Source: https://usaco.guide/general/io

#include <bits/stdc++.h>
using namespace std;
#define int long long

signed main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    int n,k;cin>>n>>k;
    vector<int> v(n);
    int fullt=0;
    vector<int> pref(n);
    for(int i = 0;i<n;i++){
        cin>>v[i];
        fullt+=v[i];
        if(i>0)pref[i]+=pref[i-1];
        pref[i]+=v[i];
    }
    for(int i = 0;i<k;i++){
        int t;cin>>t;
        if(t>0&&t%fullt==0){
            cout<<n;
            continue;
        }
        t%=fullt;
        int lo=-1,hi=n-1,mi=0;
        while(lo<hi-1){
            mi=(lo+hi+1)/2;
            if(pref[mi]<t){
                lo=mi;
            }
            else{
                hi=mi;
            }
        }
        cout<<hi+1<<'\n';
    }
}
SubtaskSumTestVerdictTimeMemory
base50/50
1Accepted0/01ms316 KiB
2Accepted0/054ms2348 KiB
3Accepted2/21ms316 KiB
4Accepted2/21ms316 KiB
5Accepted2/21ms316 KiB
6Accepted3/31ms316 KiB
7Accepted3/31ms316 KiB
8Accepted3/31ms316 KiB
9Accepted2/254ms2516 KiB
10Accepted2/256ms2356 KiB
11Accepted2/256ms2356 KiB
12Accepted2/256ms2520 KiB
13Accepted2/254ms2520 KiB
14Accepted2/254ms2548 KiB
15Accepted2/254ms2564 KiB
16Accepted2/254ms2520 KiB
17Accepted2/254ms2516 KiB
18Accepted2/254ms2356 KiB
19Accepted2/254ms2476 KiB
20Accepted2/254ms2524 KiB
21Accepted2/252ms2348 KiB
22Accepted3/352ms2572 KiB
23Accepted3/352ms2356 KiB
24Accepted3/352ms2552 KiB