题目:树的直径1
题面
时间限制:1秒 内存限制:128M
题目描述
给定一棵树,树中包含 n个结点(编号1~n)和 n−1 条无向边,每条边都有一个权值。
现在请你找到树中的一条最长路径。
要求使用广度优先遍历思想,找到一条路径,使得使得路径两端的点的距离最远。
注意:路径中可以只包含一个点。
输入描述
第一行包含整数 n。
接下来 n−1 行,每行包含三个整数 ai,bi,ci表示点 ai和 bi 之间存在一条权值为 ci 的边。
输出描述
输出一个整数,表示树的最长路径的长度。
样例输入
6
5 1 6
1 4 5
6 3 9
2 6 8
6 1 7
样例输出
22
数据范围
1≤n≤10000
1≤ai,bi≤n
0≤ci≤105
代码及思路
解题思路
- 输入及构建树
int n;
int head[20002];
int e[20002];
int ne[20002];
int w[20002];
int idx=0;
void add(int a, int b,int c) {
e[++idx]=b;
ne[idx]=head[a];
w[idx]=c;
head[a]=idx;
}
int main(){
scanf("%d",&n);
for(int i=1; i<n; i++) {
int a, b, c;
scanf("%d %d %d", &a, &b, &c);
add(a, b, c);
add(b, a, c);
}
- 深搜查找直径(步骤详见下方 查找“树的直径”的方法 )
int sum;
int node1,node2;
void dfs(int x,int father,int dis) {
if(dis>sum) {
sum=dis;
node1=x;
}
for(int i=head[x] ; i !=-1; i=ne[i]) {
if(e[i] !=father) {
dfs(e[i], x, dis+w[i]);
}
}
}
dfs(1, -1, 0);
dfs(node1,-1,0);
- 输出
printf("%d",sum);
AC代码
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
using namespace std;
int n;
int head[20002];
int e[20002];
int ne[20002];
int w[20002];
int idx=0, sum=0;
int node1,node2;
void add(int a, int b,int c) {
e[++idx]=b;
ne[idx]=head[a];
w[idx]=c;
head[a]=idx;
}
void dfs(int x,int father,int dis) {
if(dis>sum) {
sum=dis;
node1=x;
}
for(int i=head[x] ; i !=-1; i=ne[i]) {
if(e[i] !=father) {
dfs(e[i], x, dis+w[i]);
}
}
}
int main() {
memset(head,-1,sizeof head);
scanf("%d",&n);
for(int i=1; i<n; i++) {
int a, b, c;
scanf("%d %d %d", &a, &b, &c);
add(a, b, c);
add(b, a, c);
}
dfs(1, -1, 0);
dfs(node1,-1,0);
printf("%d",sum);
return 0;
}
知识点:树的直径
定义
树中最远的两个点的距离是“树的直径”。
性质
关于“树的直径”有以下性质:
性质①:直径的两个端点一定是叶子节点。
性质②:若有多条直径,则所有直径之间皆有公共点。
性质③:任取树上一个点,与之距离最远的每一个点,都是某直径的端点。
性质④:树中距离直径端点最远的点,一定是某条直径的端点,且此两个端点构成树的一条直径。
查找“树的直径”的方法
根据性质③,我们可以从根节点开始深搜,找到距离根节点最远的点,也就是树的一条直径的一个端点。再根据性质④,我们再从那个端点开始深搜,找到距离它最远的点,也就是树的那条直径的另一个端点。这样就找到树的一条直径了。
总结
这次我主要讲了以下内容:
- 题目:树的直径1。
- “树的直径”的定义。
- 关于“树的直径”的性质。
- 查找“树的直径”的方法