leetcode 133. 克隆图 基础dfs

133. 克隆图

给你无向 连通 图中一个节点的引用，请你返回该图的 深拷贝（克隆）。

图中的每个节点都包含它的值 val（int） 和其邻居的列表（list[Node]）。

class Node {
public int val;
public List neighbors; }

测试用例格式：

简单起见，每个节点的值都和它的索引相同。例如，第一个节点值为 1（val = 1），第二个节点值为 2（val =
2），以此类推。该图在测试用例中使用邻接列表表示。

邻接列表 是用于表示有限图的无序列表的集合。每个列表都描述了图中节点的邻居集。

给定节点将始终是图中的第一个节点（值为 1）。你必须将 给定节点的拷贝 作为对克隆图的引用返回。

示例 1：

---

• 两次dfs,第一次建立 节点哈希表,第二次连边
• 关于连边，注意重复的边不要连两次
• 可以优化成一次dfs(我不会)

/*
// Definition for a Node.
class Node {
public:
    int val;
    vector<Node*> neighbors;
    
    Node() {
        val = 0;
        neighbors = vector<Node*>();
    }
    
    Node(int _val) {
        val = _val;
        neighbors = vector<Node*>();
    }
    
    Node(int _val, vector<Node*> _neighbors) {
        val = _val;
        neighbors = _neighbors;
    }
};
*/

#define MAXN (128)
bool vis[MAXN];

#define cls(x) (memset(x, false, sizeof(x)))

class Solution {
public:

    void dfs(Node* u, map<int,Node*>& mp) {
        if(vis[u->val]) return ;
        vis[u->val] = true;
        mp[u->val] = new Node(u->val);
        for(int i=0; i<u->neighbors.size(); i++) {
            if(!vis[u->neighbors[i]->val]) 
                dfs(u->neighbors[i], mp);
        }
    }

    void dfs2(Node* u, Node*& p, map<int,Node*>& mp) {
#if 0
        for(auto it : mp)
            printf("%d %d\n", it.first, it.second->val);
#endif
        if(!p) p = mp[u->val];
        vis[u->val] = true;
        for(int i=0; i<u->neighbors.size(); i++) {
            if(!vis[u->neighbors[i]->val]) { //如果访问过了还连边就会重复
                Node* x = p;
                Node* y = mp[u->neighbors[i]->val];
                // printf("%d===%d\n", x->val, p->val);
                x->neighbors.push_back(y); //无向图
                y->neighbors.push_back(x);
            }
        }
        for(int i=0; i<u->neighbors.size(); i++) {
            if(!vis[u->neighbors[i]->val]) 
                dfs2(u->neighbors[i],
                     mp[p->neighbors[i]->val], mp);
        }
    }

    Node* cloneGraph(Node* node) {
        if(!node) return 0;
        cls(vis);
        map<int, Node*> mp; //值 映射 节点指针
        dfs(node, mp); //第一次dfs用来new出所有节点指针
        cls(vis);
        Node* p = 0;
        dfs2(node, p, mp); //第二次dfs用来给节点连边
        return p;
    }
};