[POI2011]Meteors

[POI2011]Meteors

luogu
SPOJ
考虑整体二分
二分国家(l_{r)和流星雨(L}R)
树状数组添加L~mid场流星雨的贡献
满足条件的国家放左边,不满足的放右边,累加贡献cur
有可能有些国家不可能满足
可以补一场inf的流星雨
复杂度:(O(nlog^2n))

#define lb(i) (i&-i)
#define ll long long
#include<bits/stdc++.h>
using namespace std;
const int _=3e5+5;
int re(){
	int x=0,w=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
	while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
	return x*w;
}
int n,m,k;
int id[_],a[_],ans[_],f[_],g[_];
ll tmp[_],cur[_],t[_];
vector<int>s[_];
struct Q{int l,r,v;}q[_];
void add(int i,int v){while(i<=m)t[i]+=v,i+=lb(i);}
ll sum(int i){ll res=0;while(i)res+=t[i],i-=lb(i);return res;}
void Add(int l,int r,int v){add(l,v);add(r+1,-v);}
void solve(int l,int r,int L,int R){//l~r:country L~R:meteors
	if(l>r)return;
	if(L==R){
		for(int i=l;i<=r;i++)ans[id[i]]=L;
		return;
	}
	int mid=(L+R)>>1;
	for(int i=L;i<=mid;i++){
		if(q[i].l<=q[i].r)Add(q[i].l,q[i].r,q[i].v);
		else Add(q[i].l,m,q[i].v),Add(1,q[i].r,q[i].v);
	}
	int x=0,y=0;
	for(int i=l;i<=r;i++){
		int u=id[i];tmp[u]=cur[u];
		for(int j=0,sz=s[u].size();j<sz;j++){
			tmp[u]+=sum(s[u][j]);
			if(tmp[u]>=a[u]){f[++x]=u;break;}
		}
		if(tmp[u]<a[u])g[++y]=u,cur[u]+=tmp[u]-cur[u];
	}
	for(int i=L;i<=mid;i++){
		if(q[i].l<=q[i].r)Add(q[i].l,q[i].r,-q[i].v);
		else Add(q[i].l,m,-q[i].v),Add(1,q[i].r,-q[i].v);
	}
	for(int i=l;i<l+x;i++)id[i]=f[i-l+1];
	for(int i=l+x;i<=r;i++)id[i]=g[i-x-l+1];
	solve(l,l+x-1,L,mid);solve(l+x,r,mid+1,R);
}
int main(){
	n=re(),m=re();
	for(int i=1;i<=m;i++)
		s[re()].push_back(i);
	for(int i=1;i<=n;i++)
		a[i]=re(),id[i]=i;
	k=re();
	for(int i=1;i<=k;i++){
		q[i].l=re(),q[i].r=re(),q[i].v=re();
	}
	q[++k]=(Q){1,m,1e9};
	solve(1,n,1,k);
	for(int i=1;i<=n;i++)
		if(ans[i]==k)puts("NIE");
		else printf("%d
",ans[i]);
	return 0;
}