https://www.luogu.org/problemnew/show/P4189
题解
模拟费用流。
首先有一个非常好的条件，每个点的限制次数都大于等于这个点的度数。
然后我们可以从\(0\)开始\(dfs\)一遍这棵树。
然后如果一条边连接的两个点的\(h\)同时\(>0\)，那么就来回走。
然后我们考虑再去从\(0\)号节点往每个节点走。
如果此时\(u\)的\(h>0\)，那么直接走就可以了。
如果不是这样，那么我们为了从\(u\)走到\(v\)，必须将上一次\(uv\)折返的路程去掉，这样会使\(v\)的次数\(+1\)，这时如果\(v\)的一个儿子的次数不为\(0\)，那么还可以继续折返一次。
代码
#include<bits/stdc++.h>
#define N 50009
using namespace std;
typedef long long ll;
int head[N],tot,h[N],ans[N],now,p[N],n;
inline ll rd(){ll x=0;char c=getchar();bool f=0;while(!isdigit(c)){if(c=='-')f=1;c=getchar();}while(isdigit(c)){x=(x<<1)+(x<<3)+(c^48);c=getchar();}return f?-x:x;
}
struct edge{int n,to;
}e[N<<1];
inline void add(int u,int v){e[++tot].n=head[u];e[tot].to=v;head[u]=tot;
}
void dfs(int u,int fa){for(int i=head[u];i;i=e[i].n)if(e[i].to!=fa){int v=e[i].to;dfs(v,u);int x=min(h[u],h[v]);now+=x*2; h[u]-=x;h[v]-=x;if(h[v])p[u]=v;}
}
void dfs2(int u,int fa){ans[u]=now;for(int i=head[u];i;i=e[i].n)if(e[i].to!=fa){int v=e[i].to;if(h[u]){h[u]--;now++;dfs2(v,u);h[u]++;now--;}else if(p[v]){h[p[v]]--;now++;dfs2(v,u);h[p[v]]++;now--; }else{h[v]++;now--;dfs2(v,u);h[p[v]]--;now++; }}
}
int main(){n=rd();int u,v;for(int i=1;i<=n;++i)h[i]=rd();for(int i=1;i<n;++i){u=rd()+1;v=rd()+1;add(u,v);add(v,u);h[u]--;h[v]--;now+=2;}dfs(1,0);dfs2(1,0);for(int i=1;i<=n;++i)printf("%d\n",ans[i]);return 0;
}
