poj 1849 树的直径（思路题）

思路：画图分析可以得出无论两个人在什么位置、怎么遍历，最后最好情况是省出了一条链只用走一次而其他的边都一定会走两次，然后就可以联想到树的直径了，如果可以只走一次树的直径，那一定是最优解。

证明：如果起点在直径上，则二人朝互相偏离的方向去遍历，则可以只走一次直径。如果起点不在直径上，则二人可以遍历完起点所在子树后（该子树每条边被访问两次），然后一起来到直径上，所以答案就是：边权和的二倍 - 树的直径。

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

const int N = 100001;
int head[N];
int n, s, e, dist, node;

struct Edge
{
    int v, w, next;
} edge[N * 2];

void addEdge( int u, int v, int w )
{
    edge[e].v = v;
    edge[e].w = w;
    edge[e].next = head[u];
    head[u] = e++;
}

void dfs( int u, int fa, int depth )
{
    if ( depth > dist )
    {
        dist = depth;
        node = u;
    }
    for ( int i = head[u]; i != -1; i = edge[i].next )
    {
        int v = edge[i].v, w = edge[i].w;
        if ( v == fa ) continue;
        dfs( v, u, depth + w );
    }
}

int main ()
{
    while ( scanf("%d%d", &n, &s) != EOF )
    {
        e = 0;
        memset( head, -1, sizeof(head) );
        int sum = 0;
        for ( int i = 1; i < n; i++ )
        {
            int u, v, w;
            scanf("%d%d%d", &u, &v, &w);
            addEdge( u, v, w );
            addEdge( v, u, w );
            sum += 2 * w;
        }
        dist = -1;
        dfs( s, -1, 0 );
        dist = -1;
        dfs( node, -1, 0 );
        sum -= dist;
        printf("%d\n", sum);
    }
    return 0;
}