POJ 2631

最新推荐文章于 2025-09-10 15:24:29 发布

转载最新推荐文章于 2025-09-10 15:24:29 发布 · 73 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/tmeteorj/archive/2012/10/10/2718815.html

文章标签：

#数据结构与算法

题意：找一颗树中的最长路。

题解：两边bfs，第一次以任意结点开始，找到一个距离它最远的结点，第二次从该节点开始，找到距离它最远的路，这边是最长路。

View Code

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<algorithm>
 4 using namespace std;
 5 const int N=10005,M=1000000;
 6 int head[N],nc;
 7 struct Edge
 8 {
 9     int to,cost,next;
10 }edge[M];
11 void add(int a,int b,int c)
12 {
13     edge[nc].to=b;edge[nc].cost=c;edge[nc].next=head[a];head[a]=nc++;
14     edge[nc].to=a;edge[nc].cost=c;edge[nc].next=head[b];head[b]=nc++;
15 }
16 int dist[N],stk[N];
17 bool vis[N];
18 int bfs(int k)
19 {
20     memset(dist,0,sizeof(dist));
21     memset(vis,false,sizeof(vis));
22     int f,r;
23     f=r=0;
24     vis[stk[r++]=k]=true;
25     int ans=0,id=k;
26     while(f!=r)
27     {
28         int now=stk[f++];
29         if(dist[now]>ans)
30             ans=dist[id=now];
31         for(int i=head[now];i!=-1;i=edge[i].next)
32         {
33             int to=edge[i].to,cc=edge[i].cost;
34             if(!vis[to])
35             {
36                 vis[to]=true;
37                 dist[to]=dist[now]+cc;
38                 stk[r++]=to;
39             }
40         }
41     }
42     return id;
43 }
44 int main()
45 {
46     int a,b,c;
47     memset(head,-1,sizeof(head));
48     nc=0;
49     while(scanf("%d%d%d",&a,&b,&c)!=EOF)
50         add(a,b,c);
51     printf("%d\n",dist[bfs(bfs(a))]);
52     return 0;
53 }