E. Turtle vs. Rabbit Race: Optimal Trainings-Codeforces Round 929 (Div. 3)

E. Turtle vs. Rabbit Race: Optimal Trainings

首先计算前缀和，对于每次询问，二分找到右边界即可。
二分：使用前缀计算区间(l,r)的总和，找到使得总和<=u的最大ansr
ansr需要考虑到，能不能取到ansr的下一位，所以要特殊判断假如取到ansr+1这个情况是否符合条件。

#include
using namespace std;
int main(){
    int T;cin>>T;
    while(T--){
        int n;cin>>n;
        int a[n+2];
        int pre[n+2]={0};
        for(int i=1;i<=n;i++) {cin>>a[i];pre[i]=pre[i-1]+a[i];}
        int q;cin>>q;
        while(q--){
            int l,u;cin>>l>>u;
            int left=l,right=n;
            int ansr=l;
            while(left<=right){
                int mid=(left+right)/2;
                if(pre[mid]-pre[l-1]<=u){
                    ansr=mid;
                    left=mid+1;
                }else{
                    right=mid-1;
                }
            }
            if(abs(pre[ansr+1]-pre[l-1]-u)<=u-(pre[ansr]-pre[l-1])) ansr++;
            cout<<ansr<<" ";
        }
        cout<<endl;
    }
}
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
26
27
28
29