请找出无向图中相连要素的个数。
图中的每个节点包含其邻居的 1 个标签和 1 个列表。(一个无向图的相连节点(或节点)是一个子图,其中任意两个顶点通过路径相连,且不与超级图中的其它顶点相连。)
样例
给定图:
A------B C
\ | |
\ | |
\ | |
\ | |
D E
返回 {A,B,D}, {C,E}
。其中有 2 个相连的元素,即{A,B,D}, {C,E}
<pre name="code" class="cpp"> * Definition for Undirected graph.
* struct UndirectedGraphNode {
* int label;
* vector<UndirectedGraphNode *> neighbors;
* UndirectedGraphNode(int x) : label(x) {};
* };
*/
class Solution {
public:
void dfs(vector<UndirectedGraphNode*> &nodes, vector<int> &path,
unordered_set<UndirectedGraphNode*> &visit, UndirectedGraphNode* n) {
visit.insert(n);
path.push_back(n->label);
for (auto &nn : n->neighbors) if (visit.find(nn) == visit.end()) { //接着访问该节点的所有子节点
dfs(nodes, path, visit, nn);
}
}
vector<vector<int>> connectedSet(vector<UndirectedGraphNode*>& nodes) {
unordered_set<UndirectedGraphNode*> visit; //把经过的节点都放在visit,为了避免重复经过
vector<vector<int>> res; //使用unordered_set是为了实现快速查找
vector<int> path; //放着同一个连通图的所有节点
for (auto &n : nodes) {
if (visit.find(n) == visit.end()) { //查找n,不在visit里面说明没有访问,那么去访问
path.clear();
dfs(nodes, path, visit, n); //dfs用递归,每次递归结束,说明访问完一个连通图,回溯机制保证每一个相连的节点都能被访问
sort(path.begin(), path.end()); //题目要求路径有序
res.push_back(path);
}
}
return res;
}
};