树的直径

Posted 2020-07-16 C-DmLy

自己写挫了...借鉴的别人的代码...

 

题目链接:http://poj.org/problem?id=2631

题解链接:http://blog.csdn.net/v5zsq/article/details/50654640

 

Description 
给出一棵无向树，求树的直径，即树上两点之间的最长距离 
Input 
每行三个整数a,b,c表示树上a点与b点之间有权值为c的一条边，以文件尾结束输入 
Output 
输出树的直径 
Sample Input 
5 1 6 
1 4 5 
6 3 9 
2 6 8 
6 1 7 
Sample Output 
22 
Solution 
先给出一个结论： 
从任一点s出发，距此点距离最远的点为直径的一个端点a，那么距a最远的点b即为直径的另一个端点，a与b之间的距离即为直径长度 
此题为求树的直径裸题，分两部分，首先从任一点出发开始dfs，不妨从1开始，用树形dp求出任一点距1点的距离，找到距其最远点s，然后以s为起点再dfs一遍，求出所有点距s的距离，其中的最大值即为树的直径 

 

 

代码:

 1 #include <cstdio>
 2 #include <cmath>
 3 #include <cstring>
 4 #include <string>
 5 #include <algorithm>
 6 #include <queue>
 7 #include <stack>
 8 #include <map>
 9 #include <set>
10 #include <vector>
11 #include <iostream>
12 using namespace std;
13 #define for0(i, n) for(int i=0; i<(n); ++i)
14 #define for1(i,a,n) for(int i=(a);i<=(n);++i)
15 #define for2(i,a,n) for(int i=(a);i<(n);++i)
16 #define for3(i,a,n) for(int i=(a);i>=(n);--i)
17 #define for4(i,a,n) for(int i=(a);i>(n);--i)
18 #define CC(i,a) memset(i,a,sizeof(i))
19 #define LL long long
20 #define MOD 1000000007
21 #define INF 0x3f3f3f3f
22 #define PI 3.1415926
23 #define N 10010
24 
25 struct Edge  
26 {  
27     Edge(){}  
28     Edge(int to,int cost,int next):to(to),cost(cost),next(next){}  
29     int to;  
30     int cost;  
31     int next;  
32 }edge[N*N];  //所有边  
33 int cnt;       //边总数  
34 int head[N];//头结点  
35 int dis[N];
36 //添加两条有向边  
37 void AddEdge(int u,int v,int cost)  
38 {  
39     edge[cnt]=Edge(v,cost,head[u]);  
40     head[u]=cnt++;  
41     edge[cnt]=Edge(u,cost,head[v]);  
42     head[v]=cnt++;  
43 }
44   
45 //BFS返回从s结点能走到的最远的点的编号  
46 int bfs(int s)  
47 {  
48     int max_dis=0;//记录最远距离  
49     int id=s;      //记录最远点  
50     queue<int> Q;  
51     memset(dis,-1,sizeof(dis));  
52     dis[s]=0;  
53     Q.push(s);  
54     while(!Q.empty())  
55     {  
56         int u=Q.front(); Q.pop();  
57         if(dis[u]>max_dis)  
58         {  
59             max_dis=dis[u];  
60             id=u;  
61         }  
62         for(int i=head[u];i!=-1;i=edge[i].next)  
63         {  
64             Edge e=edge[i];  
65             if(dis[e.to]==-1)//未访问过e.to点  
66             {  
67                 dis[e.to]=dis[u]+e.cost;  
68                 Q.push(e.to);  
69             }  
70         }  
71     }  
72     return id;  
73 }  
74   
75 int main()  
76 {  
77     int u,v,cost;  
78     memset(head,-1,sizeof(head));  
79     cnt=0;  
80     while(scanf("%d%d%d",&u,&v,&cost)==3)  
81         AddEdge(u,v,cost);  
82     printf("%d\n",dis[bfs(bfs(u))]);  
83     return 0;  
84 }