Advanced1021 Deepest Root：偷鸡小能手说的就是我吧？

最新推荐文章于 2024-09-11 02:01:15 发布

西方破晓

最新推荐文章于 2024-09-11 02:01:15 发布

阅读量104

点赞数

分类专栏： PAT习题大赏文章标签：数据结构 dfs c++

本文链接：https://blog.csdn.net/zgy_317010____/article/details/119828707

版权

PAT习题大赏专栏收录该内容

4 篇文章 0 订阅

订阅专栏

本文介绍了一种经典的深度优先搜索（DFS）题目，通过三重循环寻找图中最深的根节点集合。然而，这种方法的时间复杂度较高，可能导致超时错误。作者提出了一种优化方案，即在连续m次答案不变时停止搜索，通过调整m的值，最终在m=6时所有测试点得到正确答案。代码展示了这个优化过程，当找到唯一解时输出最深根节点，否则输出错误信息和连通分量数。

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

十分经典的DFS题目，采取了三重循环来获取最深根节点的集合。
一重循环：将每个结点都当成根节点进行DFStrave
二重循环：DFStrave
三重循环：DFS

但显然，三重循环的时间复杂度是很高的，若不加处理，我们将会在测点4获得超时报错。
于是本偷鸡小能手设置了全局变量flag，功能为：在一重循环中，DFStrave后生成的最深根节点集合连续m次不变，则视为答案以及趋于正确，不再进行循环。
在常数m的设置上，从2开始进行测试——当m较小时，会出现答案错误。测试到m=6时，获得了测点的全部正确。
（在m从2到6的变化中，6个测点轮流报错，偷鸡的快乐大大增加）

以下为代码段：

#include<iostream>
#include<set>
#include<algorithm>
using namespace std;

const int INF = 10001, m = 6;
int G[INF][INF] = { 0 }, n, con = 0, dep = 0;
bool vis[INF] = { false };
set<int> ans;
int flag = 0;

void DFS(int u, int depth) {
	vis[u] = true;
	if (dep < depth) {
		dep = depth;
		ans.clear();
		ans.insert(u);
	}
	else if (dep == depth)
		ans.insert(u);
	for (int v = 1;v <= n;v++) {
		if (vis[v] == false && G[u][v] != 0) {
			DFS(v, depth + 1);
		}
	}
}

void DFStrave(int u) {
	con = 0;
	set<int>cop;
	for (set<int>::iterator it = ans.begin();it != ans.end();it++)cop.insert(*it);
	for (int i=0;i < n;i++) {
		if (vis[(u + i - 1) % n + 1] == false) {
			con++;
			DFS((u + i - 1) % n + 1, 1);
		}
	}
	bool f = true;
	if (cop.size() == ans.size()) {
		for (set<int>::iterator it = ans.begin();it != ans.end();it++) {
			if (cop.find(*it) == cop.end()) {
				f = false;
				flag = 0;break;
			}
		}
	}
	if (f)flag++;
	fill(vis, vis + INF, false);
}

int main()
{
	cin >> n;
	int s, d;
	for (int i = 1;i < n;i++) {
		cin >> s >> d;
		G[s][d] = 1;G[d][s] = 1;
	}
	for (int i = 1;i <= n;i++) {
		DFStrave(i);
		if (flag >= m)	
			break;
	}
	if (con == 1) {
		for (set<int>::iterator it = ans.begin();it != ans.end();) {
			cout << *it;
			if (++it != ans.end())cout << endl;
		}
	}
	else
		cout << "Error: " << con << " components";
	return 0;
}