[整体二分+树状数组]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;}