CF- 975C 前缀和二分

解题思路:

假设此时在射击到第i个勇士那么就可以用sum[i-1]+这个时刻的箭数量去二分查找第一个大于它的前缀和位置，这个位置就是接下来要到达的勇士位置.sum[i]表示位置i的前缀和。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mx = 2e5 + 10;
int n,m;
ll a[mx],b[mx],sum[mx];
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%I64d",a+i),sum[i] = sum[i-1] + a[i];
    for(int i=1;i<=m;i++) scanf("%I64d",b+i);
    ll remain = 0;
    int pos = 0;
    for(int i=1;i<=m;i++){
        if(remain>b[i]){
            remain -= b[i];
            printf("%d\n",n-pos+1);
        }else{
            int k = upper_bound(sum+pos+1,sum+n+1,b[i]-remain+sum[pos]) - sum;
            if(k==n+1){
                remain = pos = 0;
                printf("%d\n",n);
            }else{
                remain = sum[k] - (b[i]-remain+sum[pos]);
                pos = k;
                printf("%d\n",n-pos+1);
            }
        }
    }
	return 0;
}