|
|
|
@ -21,19 +21,25 @@ void dfs1(int u, int fa) {
|
|
|
|
|
int v = e[i];
|
|
|
|
|
if (v == fa) continue;
|
|
|
|
|
dfs1(v, u);
|
|
|
|
|
if (du[v] == 1) // 如果v的度为1,也就是叶子
|
|
|
|
|
if (du[v] == 1)
|
|
|
|
|
f[u] += w[i];
|
|
|
|
|
else
|
|
|
|
|
// 如果v是叶子,u->v的边权可以加到f[u]中
|
|
|
|
|
// 代表着u的最大消化能力,比如样例中的4号节点,它连接着3和5两个节点,边权分别是5,10
|
|
|
|
|
// 所以它的最大消化能力=5+10=15
|
|
|
|
|
else // 如下对方节点不是叶子,比如1号节点,需要取f(1)=min(w[u][v],f(4))
|
|
|
|
|
// 当然,别忘了1也可能有其它的子节点,需要要使用f[1]+=
|
|
|
|
|
// f[u]:表示向底向上统计来的1号节点的消化能力
|
|
|
|
|
f[u] += min(f[v], w[i]);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
// 自顶向下,由父推子,先计算,后递归
|
|
|
|
|
void dfs2(int u, int fa) {
|
|
|
|
|
for (int i = h[u]; ~i; i = ne[i]) {
|
|
|
|
|
int v = e[i];
|
|
|
|
|
if (v == fa) continue;
|
|
|
|
|
if (du[u] == 1)
|
|
|
|
|
if (du[u] == 1) // 如果u是叶子 ,那么w[v][u]=w[i]是可以火力全开的
|
|
|
|
|
g[v] = f[v] + w[i];
|
|
|
|
|
else
|
|
|
|
|
else // 如果u不是叶子
|
|
|
|
|
g[v] = f[v] + min(g[u] - min(f[v], w[i]), w[i]);
|
|
|
|
|
// 先计算再递归
|
|
|
|
|
dfs2(v, u);
|
|
|
|
|