[bzoj4849]Mole Tunnels

题目描述

鼹鼠们在底下开凿了n个洞，由n-1条隧道连接，对于任意的i>1，第i个洞都会和第i/2（取下整）个洞间有一条隧
道，第i个洞内还有ci个食物能供最多ci只鼹鼠吃。一共有m只鼹鼠，第i只鼹鼠住在第pi个洞内，一天早晨，前k只
鼹鼠醒来了，而后n-k只鼹鼠均在睡觉，前k只鼹鼠就开始觅食，最终他们都会到达某一个洞，使得所有洞的ci均大
于等于该洞内醒着的鼹鼠个数，而且要求鼹鼠行动路径总长度最小。现对于所有的1<=k<=m，输出最小的鼹鼠行动
路径的总长度，保证一定存在某种合法方案。

做法

费用流模型显然。
树高是log n，而且每个点只有两个儿子 。
很容易模拟。

#include<cstdio>#include<algorithm>#define fo(i,a,b) for(i=a;i<=b;i++)using namespace std;const int maxn=300000+10,inf=1000000000;struct dong{    int x,y;} L[maxn],R[maxn],S[maxn];int p[maxn],c[maxn],a[maxn];int i,j,k,l,t,n,m,wdc,xdl,num,ans;bool czy,gjx;int read(){    int x=0,f=1;    char ch=getchar();    while (ch<'0'||ch>'9'){        if (ch=='-') f=-1;        ch=getchar();    }    while (ch>='0'&&ch<='9'){        x=x*10+ch-'0';        ch=getchar();    }    return x*f;}void update(int x){    L[x].y=R[x].y=S[x].y=inf;    int t;    if (x*2<=n){        R[x].x=S[x*2].x;        if (a[x*2]>=0) t=1;else t=-1;        R[x].y=S[x*2].y+t;    }    if (x*2+1<=n){        L[x].x=S[x*2+1].x;        if (a[x*2+1]>=0) t=1;else t=-1;        L[x].y=S[x*2+1].y+t;    }    if (c[x]>0&&L[x].y>0){        L[x].x=x;        L[x].y=0;    }    if (c[x]>0&&R[x].y>0){        R[x].x=x;        R[x].y=0;    }    if (L[x].y<S[x].y) S[x]=L[x];    if (R[x].y<S[x].y) S[x]=R[x];}void dfs(int x){    if (x>n) return;    dfs(x*2);    dfs(x*2+1);    update(x);}void updata(int x){    if (!x) return;    update(x);    updata(x/2);}void work(int x){    int y=x;    wdc=S[x].x;num=S[x].y;    czy=1;    t=0;    while (x>1){        if (a[x]<=0) t+=1;else t-=1;        if (x%2==0) gjx=1;else gjx=0;        x/=2;        if (gjx){            if (L[x].y+t<num){                num=L[x].y+t;                wdc=L[x].x;                czy=0;                xdl=x;            }        }        else{            if (R[x].y+t<num){                num=R[x].y+t;                wdc=R[x].x;                czy=0;                xdl=x;            }        }    }    ans+=num;    x=y;    c[wdc]--;    if (czy){        y=wdc;        while (y!=x){            a[y]++;            y/=2;        }        updata(wdc);    }    else{        y=x;        while (y!=xdl){            a[y]--;            y/=2;        }        y=wdc;        while (y!=xdl){            a[y]++;            y/=2;        }        updata(x);        updata(wdc);    }}int main(){    freopen("c.in","r",stdin);freopen("c.out","w",stdout);    n=read();m=read();    fo(i,1,n) c[i]=read();    fo(i,1,m) p[i]=read();    dfs(1);    fo(i,1,m){        work(p[i]);        printf("%d ",ans);    }}