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

传送门

整体二分

先二分一个答案，判断是否可行，把可行的全都扔到左边，不可行的扔到右边

判断是否可行用树状数组就行

具体细节看代码好了

整体二分细节真多……也可能是我大脑已经退化了？

 //minamoto
 #include<iostream>
 #include<cstring>
 #include<cstdio>
 #include<vector>
 #define ll long long
 using namespace std;
 #define getc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++)
 char buf[<<],*p1=buf,*p2=buf;
 inline ll read(){
     #define num ch-'0'
     char ch;bool flag=;ll res;
     while(!isdigit(ch=getc()))
     (ch=='-')&&(flag=true);
     for(res=num;isdigit(ch=getc());res=res*+num);
     (flag)&&(res=-res);
     #undef num
     return res;
 }
 const int N=;
 int n,m,k,ans[N],mb[N],st[N],top;
 ll c[N];
 vector<int> a[N];
 struct node{
     int l,r,op,id;ll k;
     node(){}
     node(int l,int r,int op,int id,ll k):l(l),r(r),op(op),id(id),k(k){}
 }q[N],q1[N],q2[N];
 inline void add(int x,ll y){
     for(;x<=m;x+=x&-x){
         if(!c[x]) st[++top]=x;
         c[x]+=y;
     }
 }
 ll query(int x){
     ll res=;
     for(;x;x-=x&-x) res+=c[x];
     return res;
 }
 void solve(int l,int r,int ql,int qr){
     if(l>r) return;
     if(ql==qr){
         for(int i=l;i<=r;++i)
         if(q[i].op==) ans[q[i].id]=ql;
         return;
     }
     int mid=ql+qr>>,p1=,p2=;
     for(int i=l;i<=r;++i){
         switch(q[i].op){
             case :{
                 ll res=;
                 for(int j=,s=a[q[i].id].size();j<s;++j){
                     res+=query(a[q[i].id][j]);
                     if(res>=q[i].k) break;
                 }
                 if(res>=q[i].k) q1[++p1]=q[i];
                 else q[i].k-=res,q2[++p2]=q[i];
                 break;
             }
             case :{
                 if(q[i].id<=mid) add(q[i].l,q[i].k),add(q[i].r+,-q[i].k),q1[++p1]=q[i];
                 else q2[++p2]=q[i];
                 break;
             }
             case :{
                 if(q[i].id<=mid) add(,q[i].k),add(q[i].r+,-q[i].k),add(q[i].l,q[i].k),q1[++p1]=q[i];
                 else q2[++p2]=q[i];
                 break;
             }
         }
     }
     while(top) c[st[top--]]=;
     for(int i=;i<=p1;++i) q[l+i-]=q1[i];
     for(int i=;i<=p2;++i) q[l+p1+i-]=q2[i];
     solve(l,l+p1-,ql,mid),solve(l+p1,r,mid+,qr);
 }
 int main(){
 //    freopen("testdata.in","r",stdin);
     n=read(),m=read();
     for(int i=;i<=m;++i) a[read()].push_back(i);
     for(int i=;i<=n;++i) mb[i]=read();
     k=read();
     for(int i=,l,r,t;i<=k;++i){
         l=read(),r=read(),t=read();
         q[i]=node(l,r,r>=l?:,i,t);
     }
     for(int i=;i<=n;++i)
     q[i+k]=node(,,,i,mb[i]);
     solve(,k+n,,k+);
     for(int i=;i<=n;++i)
     (ans[i]!=k+)?printf("%d\n",ans[i]):puts("NIE");
     return ;
 }