洛谷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[1<<21],*p1=buf,*p2=buf;
inline ll read(){
    #define num ch-'0'
    char ch;bool flag=0;ll res;
    while(!isdigit(ch=getc()))
    (ch=='-')&&(flag=true);
    for(res=num;isdigit(ch=getc());res=res*10+num);
    (flag)&&(res=-res);
    #undef num
    return res;
}
const int N=700005;
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=0;
    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==1) ans[q[i].id]=ql;
        return;
    }
    int mid=ql+qr>>1,p1=0,p2=0;
    for(int i=l;i<=r;++i){
        switch(q[i].op){
            case 1:{
                ll res=0;
                for(int j=0,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 2:{
                if(q[i].id<=mid) add(q[i].l,q[i].k),add(q[i].r+1,-q[i].k),q1[++p1]=q[i];
                else q2[++p2]=q[i];
                break;
            }
            case 3:{
                if(q[i].id<=mid) add(1,q[i].k),add(q[i].r+1,-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--]]=0;
    for(int i=1;i<=p1;++i) q[l+i-1]=q1[i];
    for(int i=1;i<=p2;++i) q[l+p1+i-1]=q2[i];
    solve(l,l+p1-1,ql,mid),solve(l+p1,r,mid+1,qr);
}
int main(){
//    freopen("testdata.in","r",stdin);
    n=read(),m=read();
    for(int i=1;i<=m;++i) a[read()].push_back(i);
    for(int i=1;i<=n;++i) mb[i]=read();
    k=read();
    for(int i=1,l,r,t;i<=k;++i){
        l=read(),r=read(),t=read();
        q[i]=node(l,r,r>=l?2:3,i,t);
    }
    for(int i=1;i<=n;++i)
    q[i+k]=node(0,0,1,i,mb[i]);
    solve(1,k+n,1,k+1);
    for(int i=1;i<=n;++i)
    (ans[i]!=k+1)?printf("%d\n",ans[i]):puts("NIE");
    return 0;
}