求树的直径

zjnu1361

树的最远距离--中高级

Time Limit: 1000MS Memory Limit: 1000K
Total Submissions: 33 Accepted: 8
Description

给定一棵树，求这棵树中距离最远的两个节点的距离。
【限制】
50%的数据满足：1<=n<=100
100%的数据满足：1<=n<=100000

Input

第一行是一个正整数n，表示这棵树的节点数。
接下来是n-1行，每行三个正整数a，b，w（1<=a,b<=n）。表示节点a和节点b之间有一条边，长度为w（1 <= w < =10000）。
数据保证一定是一棵树。不必判错。

Output

仅一个数，表示最远距离。

Sample Input

4
1 2 10
1 3 12

1 4 15

Sample Output

27

思路：琪琪说，从某点bfs找到的最远的那个点一定是直径上的一点，否则就可以有别的点代替这个点。

所以我要先bfs找到这个点，再从这个点bfs找到最远的那个点，就是树的直径啦

ps：树的直径是树上最长的一条路

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
typedef long long ll;
struct node
{
    int to, nxt, weight;
} edge[100000 * 4 + 10];
int cnt, head[100000 + 10], vis[100000 + 10];
ll dis[100000 + 10];
void add(int u, int v, int w)
{
    edge[cnt].to = v;
    edge[cnt].nxt = head[u];
    edge[cnt].weight = w;
    head[u] = cnt++;
}
int main()
{
    int n;
    scanf("%d", &n);
    int i, j, u, v, w;
    memset(head, -1, sizeof(head));
    for (i = 1; i < n; i++)
    {
        scanf("%d%d%d", &u, &v, &w);
        add(u, v, w);
        add(v, u, w);
    }
    queue<int> q;
    q.push(1);
    memset(vis, 0, sizeof(vis));
    memset(dis, 0, sizeof(dis));
    vis[1] = 1;
    int root;
    while(!q.empty())
    {
        root = q.front();
        for (i = head[root]; ~i; i = edge[i].nxt)
        {
            if(vis[edge[i].to]==0)
            {
                q.push(edge[i].to);
                vis[edge[i].to] = 1;
                dis[edge[i].to] = dis[root] + edge[i].weight;
            }
        }
        q.pop();
    }
    ll ma = 0;
    int s = 1;
    for (i = 1; i <= n; i++)
    {
        if(dis[i]>ma)
        {
            ma = dis[i];
            s = i;
        }
    }
    q.push(s);
    memset(vis, 0, sizeof(vis));
    memset(dis, 0, sizeof(dis));
    vis[s] = 1;
    while (!q.empty())
    {
        root = q.front();
        for (i = head[root]; ~i; i = edge[i].nxt)
        {
            if (vis[edge[i].to] == 0)
            {
                q.push(edge[i].to);
                vis[edge[i].to] = 1;
                dis[edge[i].to] = dis[root] + edge[i].weight;
            }
        }
        q.pop();
    }
    ma = 0;
    sort(dis + 1, dis + 1 + n);
    printf("%lld\n", dis[n]);
    return 0;
}