题目描述如下
一道前缀和的板子题,思路一目了然,甚至题目连没学过前缀和的同学都能暴力写出一个过小样例的代码,但是注意数据范围1e6,就算用上前缀和,q次询问加上暴力的n次循环(从0到n找前缀和数组中某个值是否大于sum),直接到1e12的运算量,我们不得不再一次优化算法
注意到前缀和数组是一个递增数列,且要找满足b[i]大于sum的最小值,单调加有临界点,我们很容易想到那n次循环可以用二分的方法优化到q*log(n)的复杂度,这里这道题就差不多结束了
但是,笔者在比赛的时候还是tle了,为什么呢,细节决定成败,在本题用endl换行会在输出-1时卡住,具体原因不详,但学长大佬一致认为能用'\n'别用endl,所以小细节方面也要注意
ac代码如下
#include <bits/stdc++.h>
using namespace std;
const int N=1e6+10;
#define int long long
#define endl '\n'
int n,q,sum;
int a[N],b[N];
void solve(){
cin>>sum;
int l=1,r=n;
while(l<r){
int mid=l+r>>1;
if(b[mid]>=sum) r=mid;
else l=mid+1;
}
if(b[r]>=sum) cout<<r<<endl;
else cout<<-1<<endl;
}
signed main(){
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
cin>>n>>q;
for(int i=1;i<=n;i++){
cin>>a[i];
b[i]=b[i-1]+a[i];
}
while(q--){
solve();
}
return 0;
}