BZOJ 2527 [Poi2011]Meteors [整体二分+线段树]

题意：给你n个国家，m个观测点组成的环，每个观测点所属的国家，和每个国家所需要的陨石数，总共k波陨石向范围为（L，R）中的观测点提供a个陨石。问每个国家在第几波陨石降落的时候第一次获得所需要的陨石数量。

题解：整体二分法，类似于二分答案，每次将陨石波数二分为（L，mid）（mid+1，R），用线段树维护从第1波陨石到第mid波陨石降落时观测点陨石数量，（区间更新，单点查询），暴力判断当前mid波的陨石降落时是否能使这些国家集合满足需求，能满足要求的国家分到一组，不能满足需求则分到另一组，再分别继续进行二分陨石区间。

有一个坑点就是一个国家最多会得到(3*10^5)*(3*10^5)*10^9个陨石 爆long long = =（陨石波数*国家拥有观测点数*单次落下陨石数）

AC代码：

#include<stdio.h>#include<vector>#define N 300005using namespace std;typedef long long ll;vector<ll>vt[N];ll a[N];ll need[N];ll add[N*8];ll ql[N],qr[N],qw[N];ll ans[N];ll q,n,m;ll S[N],newS[N],pos;void update(ll l,ll r,ll L,ll R,ll root,ll k){if(l<=L&&R<=r){add[root]+=k;return ;}ll mid=L+R>>1;if(mid>=r)update(l,r,L,mid,root<<1,k);else if(mid<l)update(l,r,mid+1,R,root<<1|1,k);else {update(l,mid,L,mid,root<<1,k);update(mid+1,r,mid+1,R,root<<1|1,k);}}ll query(ll x,ll L,ll R,ll root,ll ad){if(L==R)return ad+add[root];ll mid=L+R>>1;if(x<=mid)return query(x,L,mid,root<<1,ad+add[root]);else return query(x,mid+1,R,root<<1|1,ad+add[root]);}void solve(ll l,ll r,ll L,ll R){ll mid=l+r>>1;while(pos<=mid){update(ql[pos],qr[pos],0,2*m-1,1,qw[pos]);pos++;}while(pos>=mid+2){pos--;update(ql[pos],qr[pos],0,2*m-1,1,-qw[pos]);}if(l==r){for(ll i=L;i<=R;i++){ll sum=0;ll flag=1;for(ll j=0;j<vt[S[i]].size();j++){ll to=vt[S[i]][j];sum+=query(to,0,2*m-1,1,0);sum+=query(to+m,0,2*m-1,1,0);if(sum>=need[S[i]]){flag=0;break;}}if(!flag)ans[S[i]]=l;else ans[S[i]]=-1;}return ;}ll l1=L-1,l2=R+1;for(ll i=L;i<=R;i++){ll sum=0;ll flag=1;for(ll j=0;j<vt[S[i]].size();j++){ll to=vt[S[i]][j];sum+=query(to,0,2*m-1,1,0);sum+=query(to+m,0,2*m-1,1,0);if(sum>=need[S[i]]){flag=0;break;}}if(!flag)newS[++l1]=S[i];else newS[--l2]=S[i];}for(ll i=L;i<=R;i++)S[i]=newS[i];if(l1!=L-1)solve(l,mid,L,l1);if(l2!=R+1)solve(mid+1,r,l2,R);}int main(){scanf("%lld%lld",&n,&m);for(ll i=0;i<m;i++){scanf("%lld",&a[i]);vt[a[i]-1].push_back(i);}for(ll i=0;i<n;i++)scanf("%lld",&need[i]);scanf("%lld",&q);for(ll i=0;i<q;i++){scanf("%lld%lld%lld",&ql[i],&qr[i],&qw[i]);ql[i]--;qr[i]--;if(qr[i]<ql[i])qr[i]+=m;}pos=0;for(ll i=0;i<n;i++)S[i]=i;solve(0,q-1,0,n-1);for(ll i=0;i<n;i++){if(ans[i]==-1)printf("NIE\n");else printf("%lld\n",ans[i]+1);}}