首页 > 代码库 > hdu 4679 树的直径
hdu 4679 树的直径
1 /* 2 题目大意:给n个点n-1条边的树,求删除哪条边时两个树中最大的直径与边权的乘积最小。 3 树的直径(Diameter)是指树上的最长简单路。 4 直径的求法:两遍BFS (or DFS) 5 若删除的边不是直径上的那么花费为max_len*wi 6 若删除的边是直径上的那么花费为max(dp[u][2],dp[v][2])*wi 7 */ 8 #pragma comment(linker, "/STACK:16777216") 9 #include <iostream> 10 #include <cstdio> 11 #include <cstring> 12 #include <vector> 13 using namespace std; 14 15 const int maxn=100005; 16 int dep[maxn],ans[maxn],father[maxn],max_len;//ans下标对应边的编号 17 int dp[maxn][3];//j=0存i节点的到子树中的最长路,j=1存次长路,j=2存直径 18 bool vis[maxn];//标记是否为直径上的点 19 inline int max(int a,int b){return a>b?a:b;} 20 struct node 21 { 22 int id,w,v; 23 node(int id=0,int w=0,int v=0):id(id),w(w),v(v){} 24 }; 25 vector<node> f[maxn]; 26 27 void dfs_dep(int u,int pre)//找最长路 28 { 29 for(int i=0;i<f[u].size();i++) 30 { 31 int v=f[u][i].v; 32 if(v==pre) continue; 33 dep[v]=dep[u]+1; 34 father[v]=u; 35 dfs_dep(v,u); 36 } 37 return ; 38 } 39 40 void dfs(int u,int pre) 41 { 42 dp[u][0]=dp[u][1]=dp[u][2]=0; 43 for(int i=0;i<f[u].size();i++) 44 { 45 int v=f[u][i].v; 46 if(v==pre) continue; 47 dfs(v,u); 48 int temp=dp[v][0]+1; 49 if(temp>dp[u][0]) 50 { 51 dp[u][1]=dp[u][0];dp[u][0]=temp; 52 } 53 else if(temp>dp[u][1]) 54 dp[u][1]=temp; 55 dp[u][2]=max(dp[u][2],dp[v][2]);//u在子树直径边上与不再直径边上 56 } 57 dp[u][2]=max(dp[u][2],dp[u][0]+dp[u][1]);//u在子树直径中与不再直径中 58 } 59 60 void solve(int u,int pre) 61 { 62 for(int i=0;i<f[u].size();i++) 63 { 64 int v=f[u][i].v,id=f[u][i].id,w=f[u][i].w; 65 if(v==pre) continue; 66 solve(v,u); 67 //在树的直径上 68 if(vis[u] && vis[v]) 69 ans[id]=max(ans[id],w*dp[v][2]); 70 else ans[id]=w*max_len; 71 } 72 } 73 74 int main() 75 { 76 int t,i,n,icase=0,a,b,w; 77 scanf("%d",&t); 78 while(t--) 79 { 80 scanf("%d",&n); 81 for(i=1;i<=n;i++) f[i].clear(); 82 for(i=1;i<n;i++) 83 { 84 scanf("%d%d%d",&a,&b,&w); 85 f[a].push_back(node(i,w,b)); 86 f[b].push_back(node(i,w,a)); 87 } 88 int st,ed=1,temp; 89 dfs_dep(ed,-1); 90 for(i=1;i<=n;i++) 91 if(dep[ed]<dep[i]) ed=i; 92 dep[st=ed]=0; 93 dfs_dep(st,-1); 94 ed=st; 95 for(i=1;i<=n;i++) 96 if(dep[ed]<dep[i]) ed=i; 97 max_len=dep[ed]; 98 memset(vis,0,sizeof(vis)); 99 father[st]=-1;temp=ed;100 while(father[temp]!=-1)101 vis[temp]=true,temp=father[temp];102 memset(ans,0,sizeof(ans));103 dfs(st,-1);solve(st,-1);104 dfs(ed,-1);solve(ed,-1);105 temp=1;106 for(i=1;i<n;i++)107 if(ans[i]<ans[temp]) temp=i;108 printf("Case #%d: %d\n",++icase,temp);109 }110 return 0;111 }
hdu 4679 树的直径
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。