BZOJ 2527 [Poi2011]Meteors（整体二分）

【题目链接】 http://www.lydsy.com/JudgeOnline/problem.php?id=2527

【题目大意】

　　有N个成员国。现在它发现了一颗新的星球，
　　这颗星球的轨道被分为M份（第M份和第1份相邻），第i份上有第Ai个国家的太空站。
　　这个星球经常会下陨石雨。BIU已经预测了接下来K场陨石雨的情况。
　　BIU的第i个成员国希望能够收集Pi单位的陨石样本。
　　你的任务是判断对于每个国家，它需要在第几次陨石雨之后，才能收集足够的陨石。

【题解】

　　如果枚举每场陨石雨，树状数组查询每个国家是否完成收集，复杂度O(Kmlogm),难以接受
　　因此我们对答案整体二分，不同位置的答案不断划分到对应的陨石时段计算。
　　将完成任务的国家放到左边区间，未达成的放到右边区间。复杂度O(mlogmlogK).

【代码】

#include <cstdio>
#include <cstring>
#include <vector> 
using namespace std;
const int N=300010;
typedef long long LL;
int x,n,m,T,K,l[N],r[N],val[N],id[N],p[N],tmp[N],ans[N],mark[N];
LL c[N];
vector<int> v[N];
LL query(int x){LL s=0;while(x)s+=c[x],x-=x&-x;return s;}
void add(int x,int val){while(x<=m)c[x]+=val,x+=x&-x;}
void update(int x,int k){
    if(l[x]<=r[x])add(l[x],k*val[x]),add(r[x]+1,-k*val[x]);
    else{
        add(1,k*val[x]),add(r[x]+1,-k*val[x]);
        add(l[x],k*val[x]);
    }
}
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;
    while(T<=mid)update(++T,1);
    while(T>mid)update(T--,-1);
    int cnt=0,x; LL tot;
    for(int i=l;i<=r;i++){
        tot=0; x=id[i];
        for(int j=0;j<v[x].size();j++){
            tot+=query(v[x][j]);
            if(tot>=p[x])break;
        }if(tot>=p[x])mark[x]=1,cnt++;
        else mark[x]=0;
    }int l1=l,l2=l+cnt;
    for(int i=l;i<=r;i++){
        if(mark[id[i]])tmp[l1++]=id[i];
        else tmp[l2++]=id[i];
    }for(int i=l;i<=r;i++)id[i]=tmp[i];
    solve(l,l1-1,L,mid);
    solve(l1,l2-1,mid+1,R);
}
int main(){
    while(~scanf("%d%d",&n,&m)){
        for(int i=1;i<=m;i++)scanf("%d",&x),v[x].push_back(i);
        for(int i=1;i<=n;i++)scanf("%d",&p[i]);
        scanf("%d",&K);
        for(int i=1;i<=K;i++)scanf("%d%d%d",&l[i],&r[i],&val[i]);
        l[++K]=1;r[K]=m;val[K]=0x3f3f3f3f;
        for(int i=1;i<=n;i++)id[i]=i;
        solve(1,n,1,K);
        for(int i=1;i<=n;i++){
            if(ans[i]==K)puts("NIE");
            else printf("%d
",ans[i]);
        }
    }return 0;
}