POJ 1849 树的直径 Two

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

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

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

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

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

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

 const int maxn =  + ;

 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 = ; 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) == )
     {
         for(int i = ; i <= n; i++) { G[i].clear(); C[i].clear(); }

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

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

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

     return ;
 }

代码君