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<=10000
40%的数据满足1<=n<=100
100%的数据满足1<=n<=10000 1<=a,b<=n 1<=w<=10000
树形DP,用存图的方式存树,假设1为根节点后跑DFS,f[i]记录以i为根的子树上距离i最远的点,过程中要记录最远点和次远点,和即为答案的预选值。
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<iostream>
#include<cmath>
using namespace std;
const int MAXN=1010;
struct edge
{
int to,next,val;
};
edge e[MAXN*MAXN];
int f[MAXN],head[MAXN],cnt,ans;
void addedge(int u,int v,int w)
{
e[++cnt].next=head[u];
e[cnt].to=v;
e[cnt].val=w;
head[u]=cnt;
}
bool visit[MAXN];
void dfs(int x)
{
int max2=0,max1=0;
visit[x]=1;
for (int i=head[x];i;i=e[i].next)
{
int v=e[i].to;
if (!visit[v])
{
dfs(v);
max2=max(max2,f[v]+e[i].val);
if (max2>max1)
{
swap(max1,max2);
}
}
}
ans=max(ans,max2+max1);
f[x]=max1;
}
int main()
{
int n;
scanf("%d",&n);
for (int i=1;i<n;i++)
{
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
addedge(u,v,w);
addedge(v,u,w);
}
ans=0;
dfs(1);
cout<<ans;
return 0;
}