tyvj 1520 树的直径

P1520 树的直径

时间: 1000ms / 空间: 131072KiB / Java类名: Main

描述

树的直径，即这棵树中距离最远的两个结点的距离。每两个相邻的结点的距离为1，即父亲结点与儿子结点或儿子结点与父子结点之间的距离为1.有趣的是，从树的任意一个结点a出发，走到距离最远的结点b，再从结点b出发，能够走的最远距离，就是树的直径。树中相邻两个结点的距离为1。你的任务是：给定一棵树，求这棵树中距离最远的两个结点的距离。

输入格式

输入共n行
第一行是一个正整数n，表示这棵树的结点数
接下来的n-1行，每行三个正整数a,b,w。表示结点a和结点b之间有一条边，长度为w
数据保证一定是一棵树，不必判错。

输出格式

输出共一行
第一行仅一个数，表示这棵树的最远距离

测试样例1

输入

4 
1 2 10 
1 3 12 
1 4 15

输出

27

备注

10%的数据满足1<=n<=5
40%的数据满足1<=n<=100
100%的数据满足1<=n<=10000 1<=a,b<=n 1<=w<=10000Tgotmacp

题解：f[x] 表示与当前点相连的所以路径中的最长路

          g[x]表示与当前点相连的所以路径中的次长路

当前点对答案的贡献就是最长路+次长路

#include<iostream>#include<cstdio>#include<cstring>#include<cmath>#include<algorithm>#define N 100000using namespace std;int n,m;int next[N],point[N],v[N],c[N],ans,tot;int f[N],g[N];void add(int x,int y,int z){tot++; next[tot]=point[x]; point[x]=tot; v[tot]=y; c[tot]=z;tot++; next[tot]=point[y]; point[y]=tot; v[tot]=x; c[tot]=z;}int dp(int x,int fa){g[x]=0; f[x]=0;for (int i=point[x];i;i=next[i]) if (v[i]!=fa)  {  int t=c[i]+dp(v[i],x);  if (t>f[x])   g[x]=f[x],f[x]=t;  else   if (t>g[x])    g[x]=t;  }ans=max(ans,f[x]+g[x]);return f[x];}int main(){scanf("%d",&n);for (int i=1;i<n;i++) { int x,y,z; scanf("%d%d%d",&x,&y,&z); add(x,y,z); }ans=0;dp(1,0);printf("%d\n",ans);}