每日一题 寻找宜居带（dfs应用）

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]<<" ";
    
}