整体二分+洛谷P3527 [POI2011]MET-Meteors

这个题写完之后，感受到了很多收获。二分好题。
题意：给定一个环，每个节点有一个所属国家，k次事件，每次对[l,r]区间上的每个点点权加上一个值，求每个国家最早多少次操作之后所有点的点权和能达到一个值

AC码

#include<bits/stdc++.h>
using namespace std;
const int M=6e5;
#define ll long long
struct Node{int l,r,id;ll v;}q[M],p1[M],p2[M];
int n,m,k,ans[M];
ll sum[M],mx[M];
vector<ll> G[M];
void add(int x,ll v){ for(;x<=m;x+=x&-x) sum[x]+=v;}
ll ask(int x){ll anss=0;for(;x;x-=x&-x) anss+=sum[x];return anss;}
void ef(int s,int t,int l,int r){
    if(s>t)return ;
    if(l==r){
        for(int i=s;i<=t;i++)if(q[i].id>k)ans[q[i].id-k]=l;
        return ;
    }
    int mid=l+r>>1,s1=0,t1=0;
    for(int i=s;i<=t;i++){
        if(q[i].id<=mid){
            if(q[i].l<=q[i].r)add(q[i].l,q[i].v),add(q[i].r+1,-q[i].v);
            else add(1,q[i].v),add(q[i].r+1,-q[i].v),add(q[i].l,q[i].v);
            p1[++s1]=q[i];
        }else if(q[i].id<=k){
            p2[++t1]=q[i];
        }else{
            ll tmp=0;
            for(int j=0;j<G[q[i].id-k].size();j++){
                tmp+=ask(G[q[i].id-k][j]);
                if(tmp>q[i].v)break;
            }
            if(tmp>=q[i].v)p1[++s1]=q[i];
            else q[i].v-=tmp,p2[++t1]=q[i];
        }
    }
    for(int i=s;i<=t;i++)
        if(q[i].id<=mid)
            if(q[i].l<=q[i].r)add(q[i].l,-q[i].v),add(q[i].r+1,q[i].v);
            else add(1,-q[i].v),add(q[i].r+1,q[i].v),add(q[i].l,-q[i].v);
    for(int i=1;i<=s1;i++)q[s+i-1]=p1[i];
    for(int i=1;i<=t1;i++)q[s+s1+i-1]=p2[i];
    if(s1)ef(s,s+s1-1,l,mid);
    if(t1)ef(s+s1,t,mid+1,r);
}
int main(){
    scanf("%d%d",&n,&m);
    for(int x,i=1;i<=m;i++)scanf("%d",&x),G[x].push_back(i);
    for(int i=1;i<=n;i++)scanf("%d",&mx[i]);
    scanf("%d",&k);
    for(int i=1;i<=k;i++)scanf("%d%d%lld",&q[i].l,&q[i].r,&q[i].v),q[i].id=i;
    for(int i=1;i<=n;i++)q[i+k].id=i+k,q[i+k].v=mx[i];
    ef(1,k+n,1,k+1);
    for(int i=1;i<=n;i++)ans[i]<=k?printf("%d\n",ans[i]):puts("NIE");
    return 0;
}