树形dp不定根（acwing287）

题目：

有一个树形的水系，由 N−1 条河道和 N个交叉点组成。

我们可以把交叉点看作树中的节点，编号为 1∼N，河道则看作树中的无向边。

每条河道都有一个容量，连接 x与 y 的河道的容量记为 c(x,y)。

河道中单位时间流过的水量不能超过河道的容量。

有一个节点是整个水系的发源地，可以源源不断地流出水，我们称之为源点。

除了源点之外，树中所有度数为 11 的节点都是入海口，可以吸收无限多的水，我们称之为汇点。

也就是说，水系中的水从源点出发，沿着每条河道，最终流向各个汇点。

在整个水系稳定时，每条河道中的水都以单位时间固定的水量流向固定的方向。

除源点和汇点之外，其余各点不贮存水，也就是流入该点的河道水量之和等于从该点流出的河道水量之和。

整个水系的流量就定义为源点单位时间发出的水量。

在流量不超过河道容量的前提下，求哪个点作为源点时，整个水系的流量最大，输出这个最大值。

输入格式

输入第一行包含整数 T，表示共有 T 组测试数据。

每组测试数据，第一行包含整数 N。

接下来 N−1 行，每行包含三个整数 x,y,z，表示 x，y之间存在河道，且河道容量为 z。

节点编号从 1 开始。

输出格式

每组数据输出一个结果，每个结果占一行。

数据保证结果不超过 231−1。

数据范围

N≤2×105。

输入样例：

1
5
1 2 11
1 4 13
3 4 5
4 5 10

输出样例：

26

思路： 

这是一道树形dp不定根的题，做法是先随意选取一个点作为根节点，从根节点到子节点遍历一遍，算出根节点所求的值，再根据得到的根节点值，深度优先搜索，得到各个点所要求值，比较得出答案。

此题先算出根节点流量最大值，深度优先搜索，求各个点最值，比较各点最大值。

注意：

叶子节点比较特殊，需要仔细讨论。

代码：

#define _CRT_SECURE_NO_WARNINGS 
#include <iostream>
#include <queue>
#include<algorithm>
#include<cmath>
#include<map>
#include<string.h>
#include<string>
#include<cstring>
using namespace std;
const long long N =2*1e5+1000;
const long long MAX = 1e9;
vector<pair<int,int> > p[N];
int w[N];
int n, m,maxx;
void seek(int u, int f)
{
    for (int i = 0; i < p[u].size(); i++)
    {
        int v = p[u][i].first;
        if (v == f)
            continue;
        seek(v, u);
        if (p[v].size() != 1)
            w[u] += min(w[v], p[u][i].second);
        else
            w[u] += p[u][i].second;
       
    }
}
void sek(int u,int f)
{  
   
    for (int i = 0; i < p[u].size(); i++)
    {
        int v = p[u][i].first;
        if (v == f)
            continue;
        if (p[v].size() != 1)
        {
            int sum = w[u] - min(w[v], p[u][i].second);
            if (p[u].size()!=1)
                w[v] += min(sum, p[u][i].second);
            else
                w[v] += p[u][i].second;
        }
        else
        {
            int sum = w[u] - p[u][i].second;
            w[v] = min(sum, p[u][i].second);
        }
        maxx = max(maxx, w[v]);
        sek(v, u);
    }
}
int main()
{
    int t;
    cin >> t;
    while (t--)
    {
        cin >> n;
        for (int i = 1; i <= n; i++)
            p[i].clear();
        memset(w, 0, sizeof w);
        maxx = 0;
        int x, y,z;
        for (int i = 1; i < n; i++)
        {
            scanf("%d%d%d", &x, &y, &z);
            p[x].push_back({ y,z });
            p[y].push_back({ x,z });
        }
        seek(1,1);
        maxx = w[1];
        sek(1, 1);
        cout << maxx << endl;
    }
    return 0;
}