hdu4607 树的直径

对于一棵无向树：

任意点出发，找到最远点，设这个为起点，从起点出发找到最远点为终点
这条路就是直径

 

[cpp]  view plain copy

1. #include <stdio.h>  
2. #include <string.h>  
3. #include <iostream>  
4. #include <queue>  
5. #include <vector>  
6. using namespace std;  
7. #define N 100005  
8. #define INF 1<<30  
9. int n,dis[N],E;  
10. bool vis[N];  
11. vector<int>G[N];  
12. //注意点的标号是从0还是1开始的  
13. int BFS(int x)  
14. {  
15.     int i;  
16.     for(i=0;i<=n;i++)dis[i]=INF;  
17.     memset(vis,0,sizeof(vis));  
18.     queue<int>q;while(!q.empty())q.pop();  
19.   
20.     q.push(x);  
21.     dis[x]=0;  
22.     while(!q.empty())  
23.     {  
24.         int u=q.front();q.pop();vis[u]=1;  
25.         for(i=0;i<G[u].size();i++)  
26.         {  
27.             int v=G[u][i];  
28.             if(dis[v]>dis[u]+1)  
29.             {  
30.                 dis[v]=dis[u]+1;  
31.                 E=v;  
32.                 if(!vis[v])  
33.                 {  
34.                     vis[v]=1;q.push(v);  
35.                 }  
36.             }  
37.         }  
38.     }  
39.     return E;  
40. }  
41. int main(){  
42.     int T,i,que;scanf("%d",&T);  
43.     while(T--){  
44.         scanf("%d%d",&n,&que);  
45.         for(i=0;i<=n;i++)G[i].clear();  
46.         for(i=1;i<n;i++)  
47.         {  
48.             int a,b;scanf("%d %d",&a,&b);  
49.             G[a].push_back(b);  
50.             G[b].push_back(a);  
51.         }  
52.         E=BFS(1);  
53.         E=BFS(E);  
54.         while(que--)  
55.         {  
56.             int u;scanf("%d",&u);  
57.             if(u<=dis[E]+1)printf("%d\n",u-1);  
58.             else printf("%d\n",dis[E]+(u-(dis[E]+1))*2);//如果走过节点大于树直径上的点，要走回路（显然回路是走分支的部分，rt）直径+回路  
59.         }  
60.     }  
61.     return 0;  
62. }