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

2527: [Poi2011]Meteors

好坑的一题，二分答案，check成功的国家放左边，否则放右边，然后继续分治下去，树状数组维护就可以了。

然后就炸longlong了，中间记得check成功就退出。

#include<cstdio>
#include<vector>
using namespace std;
const int MAXN=3e5+5;
int n,m,q,Now,c[MAXN],que[MAXN],q1[MAXN],q2[MAXN],Ans[MAXN];
long long F[MAXN];
struct Query{int L,R,p;}Q[MAXN];
vector<int> a[MAXN];
#include<cctype>
int read(){
	int ret=0;char ch=getchar();bool f=1;
	for(;!isdigit(ch);ch=getchar()) f^=!(ch^'-');
	for(; isdigit(ch);ch=getchar()) ret=(ret<<1)+(ret<<3)+ch-48;
	return f?ret:-ret;
}
void Put(int x,int p){for(;x<=m;x+=x&-x) F[x]+=p;}
long long Get(int x){long long Sum=0;for(;x;x-=x&-x) Sum+=F[x];return Sum;}
void Updata(int x,int f){
	if(Q[x].L<=Q[x].R) Put(Q[x].L,f*Q[x].p),Put(Q[x].R+1,-f*Q[x].p);
	else Put(1,f*Q[x].p),Put(Q[x].R+1,-f*Q[x].p),Put(Q[x].L,f*Q[x].p);
}
void Solve(int s,int t,int L,int R){
	if(s>t) return;
	if(L==R){for(int i=s;i<=t;i++) Ans[que[i]]=L;return;}
	int mid=(R+L)>>1;
	while(Now<mid) Updata(++Now,1);
	while(Now>mid) Updata(Now--,-1);
	int Top_q1=0,Top_q2=0;
	for(int i=s;i<=t;i++){
		long long Sum=0;
		for(int j=0;j<a[que[i]].size()&&Sum<c[que[i]];j++) Sum+=Get(a[que[i]][j]);
		if(Sum>=c[que[i]]) q1[++Top_q1]=que[i];else q2[++Top_q2]=que[i];
	}
	for(int i=1;i<=Top_q1;i++) que[s+i-1]=q1[i];
	for(int i=1;i<=Top_q2;i++) que[s+Top_q1+i-1]=q2[i];
	Solve(s,s+Top_q1-1,L,mid);Solve(s+Top_q1,t,mid+1,R);
}
int main(){
	n=read(),m=read();Now=0;
	for(int i=1;i<=m;i++) a[read()].push_back(i);
	for(int i=1;i<=n;i++) c[i]=read(),que[i]=i;
	q=read();
	for(int i=1;i<=q;i++) Q[i]=(Query){read(),read(),read()};
	Solve(1,n,1,q+1);
	for(int i=1;i<=n;i++) if(Ans[i]==q+1) printf("NIE\n");else printf("%d\n",Ans[i]);
	return 0;
}