【BZOJ2527】【POI2011】Meteors（整体二分）

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

输入：
第一行是两个数N,M。
第二行有M个数，第i个数Oi表示第i段轨道上有第Oi个国家的太空站。
第三行有N个数，第i个数Pi表示第i个国家希望收集的陨石数量。
第四行有一个数K，表示BIU预测了接下来的K场陨石雨。
接下来K行，每行有三个数Li,Ri,Ai，表示第K场陨石雨的发生地点在从Li顺时针到Ri的区间中（如果Li<=Ri，就是Li,Li+1,…,Ri，否则就是Ri,Ri+1,…,m-1,m,1,…,Li），向区间中的每个太空站提供Ai单位的陨石样本。

输出：
N行。第i行的数Wi表示第i个国家在第Wi波陨石雨之后能够收集到足够的陨石样本。如果到第K波结束后仍然收集不到，输出NIE。

数据范围：
1<=n,m,k<=3*10^5
1<=Pi<=10^9
1<=Ai<10^9

Sample Input

3 5

1 3 2 1 3

10 5 7

3

4 2 4

1 3 1

3 5 2

Sample Output

3

NIE

1

题解：
写的第一道整体二分。感觉整体二分和CDQ分治很像，但是CDQ分治是分治所有的操作，整体二分则相当于只对查询的进行二分。不过他们都针对于离线。
如果这个题只有一个国家，那么可以二分答案，统计是否可行。这样对于一些国家一起二分也是可以的，需要用到树状数组。
每次二分流星雨次数mid，可达到的国家答案放在左边，不行的放在右边，做完后再递归解决左右的，于是相当于降了一维复杂度，每次给所有的国家划定一个答案的上（下）界，直到达到准确答案。
有两个小细节：
1.如果所有的地方都是一个国家的，统计答案时会爆long long。因此，只要CntAns>p[i]，就break，反正不影响答案。
2.有的国家最后也收集不够，所以在最后下一场所有地区，大小为inf的流星雨，只要统计答案时是inf的，就输出NIE。
代码如下：

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<math.h>
#include<vector>
#define ll long long
#define inf 0x7f7f7f7f
#define N 300005
#define lb(x) (x&(-x))
using namespace std;
ll read()
{
    ll x=0;
    char c=getchar();
    while(c<'0' || c>'9') c=getchar();
    while(c<='9' && c>='0'){x=x*10+c-'0';c=getchar();}
    return x;
}
int T,n,m,x,p[N],k,l[N],r[N],v[N],id[N];
int ans[N],tmp[N];
bool vis[N];
vector<int> a[N];
ll t[N];
void ins(int x,int v)
{
    while(x<=m)
    {
        t[x]+=v;
        x+=lb(x);
    }
}
ll ask(int x)
{
    ll ret=0;
    while(x)
    {
        ret+=t[x];
        x-=lb(x);
    }
    return ret;
}
void insert(int x,int op)
{
    if(l[x]<=r[x])
    {
        ins(l[x],op*v[x]);
        ins(r[x]+1,-op*v[x]);
    }
    else
    {
        ins(1,op*v[x]);
        ins(r[x]+1,-op*v[x]);
        ins(l[x],op*v[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) T++,insert(T,1);
    while(T>mid) insert(T,-1),T--;
    int cnt=0,now;
    ll tot;
    for(int i=l;i<=r;i++)
    {
        tot=0;now=id[i];
        for(int j=0;j<a[now].size();j++)
        {
            tot+=ask(a[now][j]);
            if(tot>=p[now]) break;
        }
        if(tot>=p[now]) vis[now]=1,cnt++;
        else vis[now]=0;
    }
    int l1=l,l2=l+cnt;
    for(int i=l;i<=r;i++)
    {
        if(vis[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()
{
    n=read(),m=read();
    for(int i=1;i<=m;i++) 
    {
        x=read();
        a[x].push_back(i);
    }
    for(int i=1;i<=n;i++) p[i]=read();
    k=read();
    for(int i=1;i<=k;i++) l[i]=read(),r[i]=read(),v[i]=read();
    l[++k]=1;r[k]=m;v[k]=inf;
    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) printf("%d\n",ans[i]);
        else puts("NIE");
    }
    return 0;
}