Luogu P3527 [POI2011]MET-Meteors 整体二分

#include<iostream>
#include<cstdio>
using namespace std;
#define ull unsigned long long
#define ll long long
#define R register ll
#define pause (for(R i=1;i<=10000000000;++i))
#define In freopen("NOIPAK++.in","r",stdin)
#define Out freopen("out.out","w",stdout)
namespace Fread {
static char B[1<<15],*S=B,*D=B;
#ifndef JACK
#define getchar() (S==D&&(D=(S=B)+fread(B,1,1<<15,stdin),S==D)?EOF:*S++)
#endif
inline ll g() {
    R ret=0,fix=1; register char ch; while(!isdigit(ch=getchar())) fix=ch==‘-‘?-1:fix;
    if(ch==EOF) return EOF; do ret=ret*10+(ch^48); while(isdigit(ch=getchar())); return ret*fix;
} inline bool isempty(const char& ch) {return (ch<=36||ch>=127);}
inline void gs(char* s) {
    register char ch; while(isempty(ch=getchar()));
    do *s++=ch; while(!isempty(ch=getchar()));
}
} using Fread::g; using Fread::gs;

namespace Luitaryi {
const int N=3e+5+10;
int n,m,k,lim,ct;
struct node {int fir,rk; ll w;}a[N],tmp[N];
#define fir(u) a[u].fir
int vr[N],nxt[N],LL[N],RR[N],W[N],ans[N];
ll c[N<<1];
inline void adde(int u,int v) {vr[++ct]=v,nxt[ct]=fir(u),fir(u)=ct;}
inline int lbt(int x) {return x&-x;}
inline void add(int pos,int inc) {for(;pos<=lim;pos+=lbt(pos)) c[pos]+=inc;}
inline ll query(int pos) { R ret=0;
    for(;pos;pos-=lbt(pos)) ret+=c[pos]; return ret;
}
inline void solve(int l,int r,int s,int t) { 
    if(l==r) {for(R i=s;i<=t;++i) ans[a[i].rk]=l; return ;}
    R md=l+r>>1,f=s-1,b=t+1;
    for(R i=l;i<=md;++i) add(LL[i],W[i]),add(RR[i]+1,-W[i]);
    for(R u=s;u<=t;++u) { R cnt=0;
        for(R i=fir(u),lim=a[u].w;i&&cnt<lim;i=nxt[i]) { R v=vr[i];
            cnt+=query(v)+query(v+m);
        } if(cnt>=a[u].w) tmp[++f]=a[u];
        else tmp[--b]=a[u],tmp[b].w-=cnt;
    } for(R i=l;i<=md;++i) add(LL[i],-W[i]),add(RR[i]+1,W[i]);
    for(R i=s;i<=f;++i) a[i]=tmp[i];
    for(R i=t;i>=b;--i) a[i]=tmp[i];    
    solve(l,md,s,f); solve(md+1,r,b,t);
}
inline void main() {
    n=g(),m=g(),lim=m<<1;
    for(R i=1,x;i<=m;++i) x=g(),adde(x,i);
    for(R i=1;i<=n;++i) a[i].w=g(),a[i].rk=i;
    k=g(); for(R i=1;i<=k;++i) LL[i]=g(),RR[i]=g(),W[i]=g(),LL[i]>RR[i]?RR[i]+=m:0;
    solve(1,k+1,1,n); for(R i=1;i<=n;++i) ans[i]==k+1?printf("NIE\n"):printf("%d\n",ans[i]);
}
}
signed main() {
    Luitaryi::main();
    return 0;
}