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

Description

---

Byteotian Interstellar Union有N个成员国。现在它发现了一颗新的星球，这颗星球的轨道被分为M份（第M份和第1份相邻），第i份上有第Ai个国家的太空站。

这个星球经常会下陨石雨。BIU已经预测了接下来K场陨石雨的情况。
BIU的第i个成员国希望能够收集Pi单位的陨石样本。你的任务是判断对于每个国家，它需要在第几次陨石雨之后，才能收集足够的陨石。

1<=n,m,k<=3*10^5
1<=Pi<=10^9
1<=Ai<10^9

Solution

---

一开始的想法是每个国家二分答案，这样显然会T
考虑整体二分。我们二分一个操作数mid，把答案小于mid和大于mid的国家分成两类递归。区间修改单点查询就用树状数组就行了

需要注意的是中途累计求和可能会爆LL，如果已经够了就要break

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <vector>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define lowbit(x) ((x)&(-(x)))

typedef long long LL;
const int INF=0x3f3f3f3f;
const int N=600005;

struct Q {int l,r; LL v;} q[N];

std:: vector <int> vec[N];

int ans[N],rec[N],a[N];
int stack[N];
int n,m,t;

LL sum[N],c[N];

int read() {
	int x=0,v=1; char ch=getchar();
	for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
	for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
	return x*v;
}

void change(int x,LL v) {
	for (;x<=m;x+=lowbit(x)) c[x]+=v;
}

void add(int x,LL v) {
	if (q[x].l<=q[x].r) {
		change(q[x].l,v);
		change(q[x].r+1,-v);
	} else {
		change(1,v);
		change(q[x].r+1,-v);
		change(q[x].l,v);
		change(m+1,-v);
	}
}

LL get(int x) {
	LL ret=0;
	for (;x;x-=lowbit(x)) ret+=c[x];
	return ret;
}

void solve(int l,int r,int L,int R) {
	if (r<l||R<L) return ;
	if (L==R) {
		rep(i,l,r) ans[rec[i]]=L;
		return ;
	}
	int mid=(L+R)>>1;
	for (;t<mid;) {
		t++; add(t,q[t].v);
	}
	for (;t>mid;) {
		add(t,-q[t].v); t--;
	}
	rep(i,l,r) {
		sum[rec[i]]=0;
		for (int j=0;j<vec[rec[i]].size();++j) {
			sum[rec[i]]+=get(vec[rec[i]][j]);
			if (sum[rec[i]]>=a[rec[i]]) break;
		}
	}
	int top=0,wjp;
	rep(i,l,r) {
		if (sum[rec[i]]>=a[rec[i]]) {
			stack[++top]=rec[i];
		}
	} wjp=top;
	rep(i,l,r) {
		if (sum[rec[i]]<a[rec[i]]) {
			stack[++top]=rec[i];
		}
	}
	rep(i,l,r) rec[i]=stack[i-l+1];
	solve(l,l+wjp-1,L,mid);
	solve(l+wjp,r,mid+1,R);
}

int main(void) {
    freopen("data.in","r",stdin);
    freopen("myp.out","w",stdout);
	n=read(),m=read();
	rep(i,1,m) {
		int x=read();
		vec[x].push_back(i);
	}
	rep(i,1,n) a[i]=read(),rec[i]=i;
	int k=read();
	rep(i,1,k) q[i].l=read(),q[i].r=read(),q[i].v=read();
	k++; q[k].l=1,q[k].r=m,q[k].v=INF;
	solve(1,n,1,k);
	rep(i,1,n) {
		if (ans[i]==k) puts("NIE");
		else printf("%d\n", ans[i]);
	}
	return 0;
}