zjnu1361
树的最远距离--中高级
Time Limit: 1000MS Memory Limit: 1000KTotal 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
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;
}