题目描述
鼹鼠们在底下开凿了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);
}
}