lintcode-找出无向图汇总的相连要素-431

请找出无向图中相连要素的个数。

图中的每个节点包含其邻居的 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;
    }
};