[换根] Accumulation Degree

题面

link
  给定一棵树，每条边有流量限制，一个结点的流量定义为将该点看为源点，最多能流出多少水(可以从不是自身的叶子结点流出)，问这棵树最大的结点流量是多少？

分析

  求出一个结点的流量，我们可以对这个结点用一遍树型dp，对于结点 $u$ 和 其儿子 $v$， 若 $v$ 是叶子结点，那么 $dp[u]+=w(u,v)$, 否则 $dp[u]+=min(dp[v],w(u,v))$。对一个结点我们可以用 $O(n)$ 时间算出其流量，但是若是用此法算出所有的结点，是$O(n^2)$ 的复杂度，会超时。
  这时候就要考虑换根的思想：先算出固定某一点为根的答案然后考虑把它的儿子换成根会发生什么样的变化，如果这个变化是比较好算的，那么我们就可考虑每个点x为根的答案都根据以它父亲为根的结果去推。
  若我们算出父亲的流量 $f[u]$, 对于 儿子的流量 $f[v]$, 若 $v$ 是叶子结点，那么 $f[u]$ 中肯定有 $w(u,v)$ 流向 $v$，$f[u]-w(u,v)$ 流向其他，那么从 $v$ 就最多可以流出去 $min(f[u]-w,w)$; 若 $v$ 不是， 那么 $f[u]$ 有 $min(w,dp[v])$ 流向 $v$， $f[u]-min(w,dp[v])$ 流向其他，那么从 $v$ 就最多可以流出去 $dp[to]+min(w,f[u]-min(w,dp[v]))$。
  那么我们需要两遍 dfs， 第一遍算出一个结点的流量，第二遍进行换根。

#include <bits/stdc++.h>
#define pb push_back

using namespace std;

typedef long long ll;
typedef pair<int, int> P;
const int maxn = 2e5 + 10;
const int INF = 0x3f3f3f3f;
const ll mod = 998244353;

struct edge
{
    int to, next, w;
}e[maxn*2];

int head[maxn], num, t, n, deg[maxn], d[maxn], f[maxn];

void add_edge(int x, int y, int w)
{
    e[++num].to = y;
    e[num].w = w;
    e[num].next = head[x];
    head[x] = num;
}

void dfs1(int x, int fa)              //求出结点1的流量
{
    d[x] = 0;
    for(int i = head[x]; i; i = e[i].next)
    {
        int to = e[i].to, w = e[i].w;
        if(to == fa)  continue;
        dfs1(to, x);
        if(deg[to] == 1)  d[x] += w;
        else d[x] += min(d[to], w);
    }
}

void dfs2(int x, int fa)                    //换根
{
    for(int i = head[x]; i; i = e[i].next)
    {
        int to = e[i].to, w = e[i].w;
        if(to == fa)  continue;
        if(deg[to] == 1)  f[to] = min(f[x]- w, w);
        else f[to] = d[to] + min(f[x] - min(w, d[to]), w);
        dfs2(to, x);
    }
}

int main()
{
    scanf("%d", &t);
    while(t--)
    {
        scanf("%d", &n);
        fill(head + 1, head + 1 + n, 0);
        fill(deg + 1, deg + 1 + n, 0);
        num = 0;

        for(int i = 1; i < n; i++)
        {
            int x, y, z;
            scanf("%d %d %d", &x, &y, &z);
            add_edge(x, y, z);  add_edge(y, x, z);
            deg[x]++;  deg[y]++;
        }

        dfs1(1, 0);
        f[1] = d[1];
        dfs2(1, 0);
        int ans = 0;
        for(int i = 1; i <= n; i++)  ans = max(ans, f[i]);
        printf("%d\n", ans);
    }
}