L2 - 2 寻找宜居带
分数 25
全屏浏览题目
切换布局
作者 JMU_ACM
单位 集美大学
给定一棵 n 个结点的树,根节点为 1 ,每条边的距离为 1 。
定义一个点的火值为:到根结点的距离 +1(根结点火值为 1 );
定义一个点的冰值为:到以该节点为根的子树的所有叶子结点的距离的最大值 +1(叶子结点冰值为1) ;
定义一个宜居带点为:冰值与火值相同的点。
请找出有多少个宜居带点,并顺序输出这些点的编号.
输入格式:
在第一行给出一个整数 n(1≤n≤105)
接下来 n−1 行,每行给出两个整数 u,v ,表示树中一条边。
输出格式:
第一行输出一个整数 ans ,表示一共有多少个宜居带。
第二行输出 ans 个数,分别表示每个宜居带的编号。(每个宜居带编号后面加一个空格)
输入样例:
4
1 2
2 3
1 4
输出样例:
在这里给出相应的输出。例如:
1
2
样例解释
四个点的火值分别为:1,2,3,2
四个点的冰值分别为:3,2,1,1
只有 2 号点是宜居带。
这一题参考了我们学院的大佬做的,一开始自己做的只能对一个点,其他全部段错误/(ㄒoㄒ)/~~
其实看到题目,我也想到了使用dfs,但是我开了一个int类二维数组来存储无向图,最后就导致了段错误,看到题目很显然的就会想到一个节点,dfs往下遍历的过程当中,去记录他的高度,和深度为了防止超时,我们需要开一个
vector<int> edge[N];
来避免发生段错误,因为二维数组他开的大小不能超过10^5
然后来说一下dfs:我们直接用样例这个例子:
其中就是,深度一定是父亲的深度加一,高度是所有孩子结点里面最高的那个加一
比如2他的深度和高度都为2(因为根节点的默认深度就是1,叶节点的默认高度就是1)
所以我们就可以写出dfs
void dfs(int son,int father){
depth[son] = depth[father]+1;//孩子结点的深度一定是父亲深度加一
high[son] = 1;//初始化为1 ,高度自带1
for(int i = 0;i<edge[son].size();i++){
if(edge[son][i]==father) continue;
else {
dfs(edge[son][i],son);
high[son] = max(high[son],high[edge[son][i]]+1);//父亲结点的深度是当前深度和孩子深度大的那一个
}
}
}
然后这边说一下为什么dfs调用的时候要从dfs(1,0)开始,因为0是整个图的虚拟结点。1是第一个结点,如果(1,2)开始,就会遍历不全
一定要加上 if(edge[son][i]==father) continue这个条件,防止遍历过程往上遍历
代码实现:
#include<bits/stdc++.h>
using namespace std;
#define N 100001
vector<int> edge[N];
int n;
int depth[N],high[N];
void dfs(int son,int father){
depth[son] = depth[father]+1;
high[son] = 1;//初始化为1 ,高度自带1
for(int i = 0;i<edge[son].size();i++){
if(edge[son][i]==father) continue;
else {
dfs(edge[son][i],son);
high[son] = max(high[son],high[edge[son][i]]+1);
}
}
}
int main()
{
cin>>n;
int x,y;
for(int i = 0;i<n-1;i++){
cin>>x>>y;
edge[x].push_back(y);
edge[y].push_back(x);
}
dfs(1,0);//0是伪父结点,因为题目从结点是从1开始
vector<int> ans;
for(int i = 1;i<=n;i++){
if(depth[i]==high[i])
ans.push_back(i);
}
cout<<ans.size()<<endl;
for(int i = 0;i<ans.size();i++)
cout<<ans[i]<<" ";
}