POJ 1849 树的直径 Two

如果一个点开始遍历一棵树再回到原点那么每条边走两次。

现在是两个人从同一点出发，那么最后遍历完以后两人离得越远越好。

最后两人所处位置的路径上的边走了一次，其他边走了两次。

要使总路程最小，两人最后停在直径两端。

所以最终答案就是总权值*2 - 树的直径

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;

const int maxn = 100000 + 10;

int n, s;
vector<int> G[maxn], C[maxn];

int len, id;

void dfs(int u, int fa, int d)
{
    if(d > len) { len = d; id = u; }
    for(int i = 0; i < G[u].size(); i++)
    {
        int v = G[u][i];
        if(v == fa) continue;
        dfs(v, u, d + C[u][i]);
    }
}

int main()
{
    while(scanf("%d%d", &n, &s) == 2)
    {
        for(int i = 1; i <= n; i++) { G[i].clear(); C[i].clear(); }

        int sum = 0;
        for(int i = 1; i < n; i++)
        {
            int u, v, d; scanf("%d%d%d", &u, &v, &d);
            sum += d * 2;
            G[u].push_back(v); C[u].push_back(d);
            G[v].push_back(u); C[v].push_back(d);
        }

        int a;
        len = -1;
        dfs(1, 0, 0);
        len = -1;
        a = id;
        dfs(a, 0, 0);

        printf("%d
", sum - len);
    }

    return 0;
}