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

题目梗概

有n个国家，需要pi个陨石。

在每个时间在某个区间会落下x个陨石。

问每个国家何时收集完陨石。

解题思路

比较明显的整体二分。

二分时间t,表示每个国家何时收集完陨石。

把时间t之前的所有陨石砸下（可以用树状数组维护），判断是否收集完然后继续二分。

#include<cstdio>
#define LL long long
#define lowbit(x) ((x)&-(x))
using namespace std;
inline int _read(){
    int num=0;char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while (ch>='0'&&ch<='9') num=num*10+ch-48,ch=getchar();
    return num;
}
const int maxn=300005;
struct jz{
    int L,R,x;
}a[maxn];
int n,m,p[maxn],k,ans[maxn],h[maxn],t[maxn];
int tot,lnk[maxn],nxt[maxn],son[maxn];
LL w[maxn];
void add(int x,int y){nxt[++tot]=lnk[x];lnk[x]=tot;son[tot]=y;}
void change(int x,int y){while (x<=m) w[x]+=y,x+=lowbit(x);}
LL query(int x){LL num=0;while(x>0) num+=w[x],x-=lowbit(x);return num;}
void work(int L1,int R1,int L2,int R2){
    if (L2>R2) return;
    if (L1==R1){for (int i=L2;i<=R2;i++) ans[h[i]]=L1;return;}
    int mid=L1+R1>>1;
    for (int i=L1;i<=mid;i++)
    if (a[i].L<=a[i].R) change(a[i].L,a[i].x),change(a[i].R+1,-a[i].x);else
    change(a[i].L,a[i].x),change(1,a[i].x),change(a[i].R+1,-a[i].x);
    int i1=L2-1,i2=R2+1;
    for (int i=L2;i<=R2;i++){
        LL num=0;
        for (int j=lnk[h[i]];j;j=nxt[j]){
            num+=query(son[j]);
            if (num>=p[h[i]]) break;
        }
        if (num>=p[h[i]]) t[++i1]=h[i];else t[--i2]=h[i],p[h[i]]-=num;
    }
    for (int i=L2;i<=R2;i++) h[i]=t[i];
    for (int i=L1;i<=mid;i++)
    if (a[i].L<=a[i].R) change(a[i].L,-a[i].x),change(a[i].R+1,a[i].x);else
    change(a[i].L,-a[i].x),change(1,-a[i].x),change(a[i].R+1,a[i].x);
    work(L1,mid,L2,i1);work(mid+1,R1,i1+1,R2);
}
int main(){
    freopen("exam.in","r",stdin);
    freopen("exam.out","w",stdout);
    n=_read(),m=_read();
    for (int i=1;i<=m;i++){int x=_read();add(x,i);}
    for (int i=1;i<=n;i++) p[i]=_read(),h[i]=i;
    k=_read();
    for (int i=1;i<=k;i++) a[i].L=_read(),a[i].R=_read(),a[i].x=_read();
    a[k+1].L=1;a[k+1].R=m;a[k+1].x=1e9;
    work(1,k+1,1,n);
    for (int i=1;i<=n;i++)
    if (ans[i]>k) printf("NIE\n");else printf("%d\n",ans[i]);
    return 0;
}