改进以下代码#include <iostream> #include <cstring> #include <algorithm> #include <set> #include <vector> using namespace std; const int maxn = 1010; set<int> Adj[maxn]; bool hs[maxn]; vector<int> path; int Nv,Ne,st; void Read() { cin >> Nv >> Ne >> st; for(int i=0;i<Ne;++i) { int u,v; cin >> u >> v; Adj[u].insert(v); Adj[v].insert(u); } } void dfs(int u) { hs[u] = 1; cout << u << ' '; for(auto v:Adj[u]) if(hs[v]==0) { path.push_back(v); //将这个节点加到path中 dfs(v); //递归 path.pop_back(); //递归完成以后,需要返回,这时就需要打印 // 返回时结点的信息,先删除尾元素,再打印 if(path.size()) //要判断path是否为空 cout << path.back() << ' '; } } int main() { Read(); dfs(st); int cnt=0; for(int i=1;i<=Nv;++i) cnt += hs[i]; //判断地图是否连通 cout << st ; if(cnt != Nv) puts(" 0"); return 0; }
时间: 2024-02-16 17:26:37 浏览: 68
使用include <iostream>时一定要加using namespace std
这段代码是基于深度优先搜索(DFS)实现的地图遍历,对于小规模的地图遍历问题是可以使用的,但是若遇到大规模的地图遍历问题则会存在效率问题,因为该算法没有使用剪枝等优化方法,因此可能会出现重复遍历的情况。同时,该代码也没有进行错误处理,例如当输入的节点数或边数超过最大值时程序可能会崩溃。以下是对该代码的改进建议:
1. 使用邻接表存储图
该代码使用的是set存储图的节点,但是对于大规模的图来说,这种存储方式的效率并不高。可以使用邻接表来存储图,避免重复存储边信息,提高程序效率。
2. 使用剪枝优化
在DFS的过程中,可以使用剪枝来避免重复遍历相同的节点,提高程序效率。具体来说,可以使用一个布尔数组记录已经遍历过的节点,遇到已经遍历过的节点就跳过,避免重复遍历。
3. 添加错误处理
在读入节点数和边数时,应该进行错误处理,例如当输入的节点数或边数超过最大值时应该给出提示并退出程序。
4. 修改函数参数
在DFS函数中,应该将path参数传入函数中,而不是使用全局变量,这样可以避免在多线程环境下出现问题。
5. 简化代码逻辑
在DFS函数中,可以将打印节点信息的逻辑简化,避免使用额外的vector来记录遍历路径。可以在递归的过程中传递已遍历的节点信息,然后在返回时打印即可。
综上所述,以下是改进后的代码:
阅读全文