标签:gif dfs open space alt printf oid lan style
题目链接:http://lx.lanqiao.cn/problem.page?gpid=T32
题意:中文题诶~
思路:显然给出的地图是一颗树,若能求得树的直径 ans,则答案为:ans*(ans+1)/2 + 10*ans;
对于树的直径:若已知根节点root,则 ans 为 root 高度最大的两颗子树高度和,显然我们可以先dfs一遍找到 root 最高的子树并确定离 root 最远的那个节点 v ;
然后再以 v 为根节点,显然 v 只有一颗子树,那么这颗子树的高度即为此树的直径.只需从 v 出发再dfs一遍即可 ;
所以对于求树的直径我们可以先以任意一点为出发点dfs出距离此节点最远的节点v,再从v出发dfs一遍距离v最远的点,此时求出的距离即为直径;
代码:
1 #include <stdio.h> 2 #include <vector> 3 #include <algorithm> 4 #include <string.h> 5 using namespace std; 6 7 vector<pair<int,int> >tree[100005]; 8 bool vis[100005]; 9 int end_root,Max_len; 10 11 void dfs(int x,int len){ 12 vis[x]=true; 13 if(len>Max_len) Max_len=len,end_root=x; 14 for(int i=0;i<tree[x].size();i++){ 15 if(!vis[tree[x][i].first]){ 16 dfs(tree[x][i].first,len+tree[x][i].second); 17 } 18 } 19 } 20 21 int main(void){ 22 int n; 23 while(~scanf("%d",&n)){ 24 int max_val=0; 25 memset(tree,0,sizeof(tree)); 26 for(int i=1;i<n;i++){ 27 int a,b,c; 28 scanf("%d %d %d",&a,&b,&c); 29 tree[a].push_back({b,c}); 30 tree[b].push_back({a,c}); 31 } 32 Max_len=0; 33 memset(vis,0,sizeof(vis)); 34 dfs(1,0); 35 memset(vis,0,sizeof(vis)); 36 dfs(end_root,0); 37 long long ans=Max_len; 38 printf("%lld\n",ans*(1+ans)/2+10*ans); 39 } 40 }
标签:gif dfs open space alt printf oid lan style
原文地址:http://www.cnblogs.com/geloutingyu/p/6871818.html