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

题目传送门
似乎是个整体二分板题（然而是300000的nlog²n。。）？
先把前mid场陨石雨插到树状数组里面。
对每个国家开个vector，统计它收集到的陨石数（注意到轨道的总数是O(n)的）

才打几道整体二分，注意几个坑点：
分到右边的询问要减去左边对它的贡献。
复制的tmp数组的下标不能用tmp[0]，因为分左右的时候会被改，要用局部变量。

Code：

#include<cstdio>
#include<cctype>
#include<vector>
#include<cstring>
#include<algorithm>
#define LL long long
#define maxn 300005
using namespace std;
char cb[1<<15],*cs,*ct;
#define getc() (cs==ct&&(ct=(cs=cb)+fread(cb,1,1<<15,stdin),cs==ct)?0:*cs++)
inline void read(int &a){
	char c;while(!isdigit(c=getc()));
	for(a=c-'0';isdigit(c=getc());a=a*10+c-'0');
}
int n,m,K,L[maxn],R[maxn],P[maxn],need[maxn],ans[maxn],id[maxn],ql[maxn],qr[maxn];
LL arr[maxn],sum[maxn];
void update(int i,int d){for(;i<=m;i+=i&-i) arr[i]+=d;}
LL getsum(int i){LL s=0;for(;i;i-=i&-i) s+=arr[i];return s;}
vector<int>a[maxn];
void solve(int h,int t,int l,int r){
	if(h>t) return;
	if(l==r) {for(int i=h;i<=t;i++) ans[id[i]]=l;return;}
	int mid=(l+r)>>1;
	for(int i=l;i<=mid;i++){
		if(L[i]<=R[i]) update(L[i],P[i]),update(R[i]+1,-P[i]);
		else update(1,P[i]),update(R[i]+1,-P[i]),update(L[i],P[i]);
	}
	int ls=0,rs=0;
	for(int i=h;i<=t;i++){
		for(int j=a[id[i]].size()-1;j>=0;j--) if((sum[id[i]]+=getsum(a[id[i]][j]))>=need[id[i]]) break;
		if(sum[id[i]]>=need[id[i]]) ql[++ls]=id[i];
		else need[id[i]]-=sum[id[i]],qr[++rs]=id[i];
		sum[id[i]]=0;
	}
	for(int i=l;i<=mid;i++){
		if(L[i]<=R[i]) update(L[i],-P[i]),update(R[i]+1,P[i]);
		else update(1,-P[i]),update(R[i]+1,P[i]),update(L[i],-P[i]);
	}
	memcpy(id+h,ql+1,ls<<2);
	memcpy(id+h+ls,qr+1,rs<<2);
	solve(h,h+ls-1,l,mid);
	solve(h+ls,t,mid+1,r);
}
int main()
{
	#ifndef ONLINE_JUDGE
		freopen("H.in","r",stdin);
	#endif
	read(n),read(m);
	for(int i=1,x;i<=m;i++) read(x),a[x].push_back(i);
	for(int i=1;i<=n;i++) read(need[i]);
	read(K);
	for(int i=1;i<=K;i++) read(L[i]),read(R[i]),read(P[i]);
	for(int i=1;i<=n;i++) id[i]=i;
	solve(1,n,1,K+1);
	for(int i=1;i<=n;i++) 
		if(ans[i]<=K) printf("%d\n",ans[i]);
		else puts("NIE");
}