P1520 树的直径
时间: 1000ms / 空间: 131072KiB / Java类名: Main
描述
树的直径,即这棵树中距离最远的两个结点的距离。每两个相邻的结点的距离为1,即父亲结点与儿子结点或儿子结点与父子结点之间的距离为1.有趣的是,从树的任意一个结点a出发,走到距离最远的结点b,再从结点b出发,能够走的最远距离,就是树的直径。树中相邻两个结点的距离为1。你的任务是:给定一棵树,求这棵树中距离最远的两个结点的距离。
输入格式
输入共n行
第一行是一个正整数n,表示这棵树的结点数
接下来的n-1行,每行三个正整数a,b,w。表示结点a和结点b之间有一条边,长度为w
数据保证一定是一棵树,不必判错。
第一行是一个正整数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
40%的数据满足1<=n<=100
100%的数据满足1<=n<=10000 1<=a,b<=n 1<=w<=10000Tgotmacp
g[x]表示与当前点相连的所以路径中的次长路
当前点对答案的贡献就是最长路+次长路
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define N 100000
using 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);
}