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

题目传送门

这题的整体二分还是挺好想到的，代码也挺好写的。（比BZOJ3110好写多……）

取当期的陨石波数的一半加入树状数组中，判断当前每个国家拥有的陨石数是否达到需求量，若达到则放到左区间中，也就是减小陨石数，反之则放到右区间，增加陨石数。

orzZH大佬，当讲课人在上面放出了这道题时，ZH大佬瞟了一眼，哼了一声：“这题不是水题吗？还需要讲？”

ZH大佬太强啦QWQ，%%%%%，献上我的膝盖骨——跪膜大佬。

附上AC代码：

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

struct side{int to,nt;}s[N];
struct note{long long w,wz;}a[N],q1[N],q2[N];
struct ask{long long l,r,w;}q[N];
long long n,m,x,h[N],num,p,ans[N],t[N];

inline char nc(){
	static char ch[100010],*p1=ch,*p2=ch;
	return p1==p2&&(p2=(p1=ch)+fread(ch,1,100010,stdin),p1==p2)?EOF:*p1++;
}

inline void read(long long &a){
	static char ch=nc();int f=1;
	for (;!isdigit(ch);ch=nc()) if (ch=='-') f=-1;
	for (a=0;isdigit(ch);a=a*10+ch-'0',ch=nc());
	a*=f;return;
}

inline void add(int x,int y){s[++num]=(side){y,h[x]},h[x]=num;}
inline void updata(int x,long long w){for (int i=x; i<=n; i+=lowbit(i)) t[i]+=w;}

inline void change(int now,int ww){
	for (int i=now; i<=m; i+=lowbit(i))
		t[i]+=ww;
}

inline void updata(int l,int r,long long w){
	if (l>r) change(l,w),change(1,w),change(r+1,-w);
		else change(l,w),change(r+1,-w);
}

inline long long query(int now){
	long long ret=0;
	for (int i=now; i; i-=lowbit(i))
		ret+=t[i];
	return ret;
}

inline void so(int l,int r,int ql,int qr){
	if (l>r) return;
	if (ql==qr){
		for (int i=l; i<=r; ++i) ans[a[i].wz]=ql;
		return;
	}
	int mid=(ql+qr)>>1,ll=0,rr=0,len=l;
	for (int i=ql; i<=mid; ++i) updata(q[i].l,q[i].r,q[i].w);
	for (int i=l; i<=r; ++i){
		long long sum=0;
		for (int j=h[a[i].wz]; j; j=s[j].nt){
			sum+=query(s[j].to);
			if (sum>=a[i].w) break;
		}
		if (sum>=a[i].w) q1[++ll]=a[i];
			else q2[++rr]=a[i],q2[rr].w-=sum;
	} 
	for (int i=1; i<=ll; ++i) a[len++]=q1[i];
	for (int i=1; i<=rr; ++i) a[len++]=q2[i];
	for (int i=ql; i<=mid; ++i) updata(q[i].l,q[i].r,-q[i].w);
	return so(l,l+ll-1,ql,mid),so(l+ll,r,mid+1,qr);
}

int main(void){
	read(n),read(m);
	for (int i=1; i<=m; ++i) read(x),add(x,i);
	for (int i=1; i<=n; ++i) read(a[i].w),a[i].wz=i;
	read(p);
	for (int i=1; i<=p; ++i) read(q[i].l),read(q[i].r),read(q[i].w);
	so(1,n,1,p+1);
	for (int i=1; i<=n; ++i) if (ans[i]>p) puts("NIE"); else printf("%d\n",ans[i]);
	return 0;
}