PAT1021 Deepest Root (25)(树的最深节点)

最新推荐文章于 2022-03-06 17:13:01 发布

SeasonJoe

最新推荐文章于 2022-03-06 17:13:01 发布

阅读量346

点赞数

本文链接：https://blog.csdn.net/SeasonJoe/article/details/79830032

版权

该题目要求判断给定边是否能构成一棵树，并找到作为根时深度最大的节点。解决方案是首先以1为根进行深度优先搜索（DFS），找出最深的节点集合。然后选择其中一个最深节点再次进行DFS，找到的节点集合与之前的集合合并，这些节点即为以它们为根时树的最深节点。

摘要由CSDN通过智能技术生成

题意：

给出n-1条边，要求判断能否生成一棵树，并求树中所有以它为根时深度最大的节点。

思路：

我一开始做的时候就很莽的直接用dfs搜索，就有一个节点超时，其实简单的思考就是如果以1为根，那么就是当前最深的几个点，他们之间的距离也就是树最大的深度。可以这么理解：从1到i距离最远，那么以i为根，到1的距离已经是最远，那只要从1开始找次深的点即可。实现方法就是先dfs一遍找出最深的几个点，在用其中一个再次dfs，找到的点的集合和前一个集合的并集就是答案。

#include<string>
#include<cstdlib>
#include<vector>
#include<stack>
#include<queue>
#include<utility>
#include<map>
#include<set>
#include<cstdlib>
#include<stdio.h>
#include<algorithm>
#include<functional>
#include<iostream>
#define INF 0x6ffff
using namespace std;

const int maxn = 10010;
vector<int> edge[maxn];
vector<int> temp;
set<int> res;
bool vis[maxn];
int n;
int deepMax;

void dfs(int u, int depth) {
	if (depth > deepMax) {
		deepMax = depth;
		temp.clear();
		temp.push_back(u);
	}
	else