[Poi2011]Meteors 【整体二分+树状数组】

传送门（内校oj）

---

题面：

---

SOL

整体二分即可，二分第几波陨石

（各种写挂。。。以后不复制粘贴了）

---

CODE

代码有点丑。。

#include<bits/stdc++.h>
using namespace std;
#define sf scanf
#define pf printf
#define ll long long
const int N=3e5+10;
struct node{
	int l,r,k,id,op;
	//1:query 2:update : tim:country
}q[N*3],q1[N*3],q2[N*3];
int ans[N],n,m,K,cnt;
ll req[N];
vector<int> G[N];
struct tree{
	#define lb(k) (k&(-k))
	ll sum[N];
	inline void init(){memset(sum,0,sizeof sum);}
	inline void update(int p,int k){
		for(;p<=m;p+=lb(p))sum[p]+=k;
	}
	inline ll query(int p){
		ll res=0;	
		for(;p>=1;p-=lb(p))res+=sum[p];
		return res;
	}
}tr;
void solve(int l,int r,int st,int ed){
	if(l==r){
		for(int i=st;i<=ed;++i)if(q[i].op)ans[q[i].id]=l;
		return;
	}
	int mid=(l+r)>>1,cnt1=0,cnt2=0;
	for(int i=st;i<=ed;++i){
		if(q[i].op){
//			cout<<"id:"<<q[i].id<<'\n';
			ll res=0;
			for(int j=0;j<G[q[i].id].size();++j){
				res+=tr.query(G[q[i].id][j]);
				if(res>=req[q[i].id])break;
			}
//			cout<<req[q[i].id]<<"->"<<res<<'\n';
			if(res>=req[q[i].id])q1[++cnt1]=q[i];
			else req[q[i].id]-=res,q2[++cnt2]=q[i];
		}
		else{
			if(q[i].id<=mid)tr.update(q[i].l,q[i].k),tr.update(q[i].r+1,-q[i].k),q1[++cnt1]=q[i];
			else q2[++cnt2]=q[i];
		}
	}
	for(int i=1;i<=cnt1;++i)if(!q1[i].op)tr.update(q1[i].l,-q1[i].k),tr.update(q1[i].r+1,q1[i].k);
	for(int i=1;i<=cnt1;++i)q[st+i-1]=q1[i];
	for(int i=1;i<=cnt2;++i)q[st+cnt1+i-1]=q2[i];
	solve(l,mid,st,st+cnt1-1);solve(mid+1,r,st+cnt1,ed);
}
signed main(){
	sf("%d%d",&n,&m);
	for(int i=1;i<=m;++i){
		int k;sf("%d",&k);
		G[k].push_back(i);
	}	
	for(int i=1;i<=n;++i){
		sf("%lld",&req[i]);
	}
	sf("%d",&K);
	for(int i=1;i<=K;++i){
		int l,r,k;sf("%d%d%d",&l,&r,&k);
		if(l<=r){
			q[++cnt]=(node){l,r,k,i,0};
			tr.update(l,k);
			tr.update(r+1,-k);
		}
		else{
			swap(l,r);
			q[++cnt]=(node){r,m,k,i,0};
			q[++cnt]=(node){1,l,k,i,0};
			tr.update(r,k);
			tr.update(1,k);
			tr.update(l+1,-k);
		}
	}
	for(int i=1;i<=n;++i){
		ll res=0;
		for(int j=0;j<G[i].size();++j){
			res+=tr.query(G[i][j]);
			if(res>=req[i])break;
		}
		if(res<req[i])ans[i]=-1;
		else q[++cnt]=(node){0,0,0,i,1};
	}
	tr.init();
	solve(1,K,1,cnt);
	for(int i=1;i<=n;++i){
		if(ans[i]^(-1))cout<<ans[i]<<'\n';
		else cout<<"NIE\n";
	}

	return 0;
}