[BZOJ2527] [Poi2011]Meteors（整体二分）

对于单个国家，可以对答案进行二分，每次找出此时的陨石数量，如果大于需要的那么答案就在[l,mid]，否则就在[mid+1,r]里面

而对于很多国家，也可以进行二分，solve(l,r,L,R)表示询问id[l...r]的答案都在[L,R]之间

每次用树状数组统计一下

Code

#include <cstdio>
#include <algorithm>
#define lowbit(x) ((x)&(-x))
#define N 300010
#define ll long long
using namespace std;

struct edge{int to,nex;}e[N*2];
struct que{int x,y,z;}q[N];
const int Inf=1e9;
int n,m,k,tot,head[N],A[N],id[N],Ans[N],tmp,tmp2[N];
ll T[N];

inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

inline void Link(int u,int v){
	e[++tot].to=v;e[tot].nex=head[u];head[u]=tot;
}

ll query(int x){
	ll res=0;
	for(;x;x-=lowbit(x)) res+=T[x];
	return res;
}

void add(int x,int v){
	for(;x<=m;x+=lowbit(x)) T[x]+=v;
}

void opera(int k,int f){
	add(q[k].x,f*q[k].z),add(q[k].y+1,f*q[k].z*(-1));
	if(q[k].x>q[k].y) add(1,f*q[k].z);
}

bool mk[N];
inline void solve(int l,int r,int L,int R){
	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(;tmp<=mid;) opera(++tmp,1);
	for(;tmp>mid;) opera(tmp--,-1);
	int cnt=0,u;ll tot;
	for(int i=l;i<=r;++i){
		tot=0,u=id[i];
		for(int j=head[u];j;j=e[j].nex){
			int v=e[j].to;
			tot+=query(v);
			if(tot>=A[u]) break;
		}
		if(tot>=A[u]) mk[u]=1,cnt++;
		else mk[u]=0;
	}
	int l1=l,l2=l+cnt;
	for(int i=l;i<=r;++i)
		if(mk[id[i]]) tmp2[l1++]=id[i];
		else tmp2[l2++]=id[i];
	for(int i=l;i<=r;++i) id[i]=tmp2[i];
	solve(l,l1-1,L,mid);
	solve(l1,l2-1,mid+1,R);
}

int main(){
	n=read(),m=read();
	for(int i=1,x;i<=m;++i) x=read(),Link(x,i);
	for(int i=1;i<=n;A[i++]=read());
	k=read();
	for(int i=1;i<=k;++i) q[i].x=read(),q[i].y=read(),q[i].z=read();
	q[++k].x=1,q[k].y=m,q[k].z=Inf;
	for(int i=1;i<=n;++i) id[i]=i;
	solve(1,n,1,k);
	for(int i=1;i<=n;++i) (Ans[i]<k)?printf("%d\n",Ans[i]):puts("NIE");
	return 0;
}