133. 克隆图

链接:133. 克隆图

题解:

深度优先搜索
思路

对于本题而言,我们需要明确图的深拷贝是在做什么,对于一张图而言,它的深拷贝即构建一张与原图结构,值均一样的图,但是其中的节点不再是原来图节点的引用。因此,为了深拷贝出整张图,我们需要知道整张图的结构以及对应节点的值。

由于题目只给了我们一个节点的引用,因此为了知道整张图的结构以及对应节点的值,我们需要从给定的节点出发,进行「图的遍历」,并在遍历的过程中完成图的深拷贝。

为了避免在深拷贝时陷入死循环,我们需要理解图的结构。对于一张无向图,任何给定的无向边都可以表示为两个有向边,即如果节点 A 和节点 B 之间存在无向边,则表示该图具有从节点 A 到节点 B 的有向边和从节点 B 到节点 A 的有向边。

为了防止多次遍历同一个节点,陷入死循环,我们需要用一种数据结构记录已经被克隆过的节点。

算法

使用一个哈希表存储所有已被访问和克隆的节点。哈希表中的 key 是原始图中的节点,value 是克隆图中的对应节点。

从给定节点开始遍历图。如果某个节点已经被访问过,则返回其克隆图中的对应节点。

如下图,我们给定无向边边 A - B,表示 A 能连接到 B,且 B 能连接到 A。如果不对访问过的节点做标记,则会陷入死循环中。

如果当前访问的节点不在哈希表中,则创建它的克隆节点并存储在哈希表中。注意:在进入递归之前,必须先创建克隆节点并保存在哈希表中。如果不保证这种顺序,可能会在递归中再次遇到同一个节点,再次遍历该节点时,陷入死循环。


递归调用每个节点的邻接点。每个节点递归调用的次数等于邻接点的数量,每一次调用返回其对应邻接点的克隆节点,最终返回这些克隆邻接点的列表,将其放入对应克隆节点的邻接表中。这样就可以克隆给定的节点和其邻接点。

/*
// 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;
    }
};
*/

class Solution {
public:
    unordered_map<Node*, Node*> visited;
    Node* cloneGraph(Node* node) {
        if (node == nullptr) {
            return node;
        }

        // 如果该节点已经被访问过了,则直接从哈希表中取出对应的克隆节点返回
        if (visited.find(node) != visited.end()) {
            return visited[node];
        }

        // 克隆节点,注意到为了深拷贝我们不会克隆它的邻居的列表
        Node* cloneNode = new Node(node->val);
        // 哈希表存储
        visited[node] = cloneNode;

        // 遍历该节点的邻居并更新克隆节点的邻居列表
        for (auto& neighbor: node->neighbors) {
            cloneNode->neighbors.emplace_back(cloneGraph(neighbor));
        }
        return cloneNode;
    }
};
/*
// 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;
    }
};
*/

class Solution {
public:
    Node* cloneGraph(Node* node) {
        // 找到所有节点
        std::vector<Node*> nodes = find_nodes(node);
        // 复制所有新节点
        std::unordered_map<Node*, Node*> mapping_nodes = clone_nodes(nodes);
        // 建立新的连接关系
        build_edges(nodes, mapping_nodes);
        // 返回新图节点
        return mapping_nodes[node];
    }
private:
    std::vector<Node*> find_nodes(Node* node) {
        if (!node) {
            return {};
        }
        std::queue<Node*> que;
        que.push(node);
        std::unordered_set<Node*> visited;
        visited.insert(node);
        std::vector<Node*> result;
        result.push_back(node);
        while (!que.empty()) {
            auto f = que.front();
            que.pop();
            for (auto neighbor : f->neighbors) {
                if (visited.find(neighbor) != visited.end()) {
                    continue;
                }
                result.push_back(neighbor);
                que.push(neighbor);
                visited.insert(neighbor);
            }
        }
        return result;
    }
    std::unordered_map<Node*, Node*> clone_nodes(const std::vector<Node*>& nodes) {
        std::unordered_map<Node*, Node*> clones;
        for_each(nodes.begin(), nodes.end(), [&clones](auto node) {
            clones[node] = new (std::nothrow) Node(node->val);
        });
        return clones;
    }
    void build_edges(std::vector<Node*>& nodes, std::unordered_map<Node*, Node*>& mapping_nodes) {
        for_each(nodes.begin(), nodes.end(), [&mapping_nodes](auto node) {
            auto mapping_node = mapping_nodes[node];
            for (auto neighbor : node->neighbors) {
                mapping_node->neighbors.push_back(mapping_nodes[neighbor]);
            }
        });
    }
};

/*
// 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;
    }
};
*/

class Solution {
public:
    Node* cloneGraph(Node* node) {
        if (!node) {
            return nullptr;
        }
        std::unordered_map<Node*, Node*> table;
        copy_node(node, table);
        connect_node(node, table);
        return table[node];
    }
private:
    void copy_node(Node* node, std::unordered_map<Node*, Node*>& table) {
        if (!node) {
            return;
        }
        std::queue<Node*> que;
        que.push(node);
        std::unordered_set<Node*> visited;
        visited.insert(node);
        while (!que.empty()) {
            auto f = que.front();
            table[f] = new (std::nothrow)Node(f->val);
            que.pop();
            for (auto neigh : f->neighbors) {
                if (visited.find(neigh) != visited.end()) {
                    continue;
                }
                que.push(neigh);
                visited.insert(neigh);
            }
        }
    }
    void connect_node(Node* node, std::unordered_map<Node*, Node*>& table) {
        if (!node) {
            return;
        }
        std::queue<Node*> que;
        que.push(node);
        std::unordered_set<Node*> visited;
        visited.insert(node);
        while (!que.empty()) {
            auto f = que.front();
            que.pop();
            for (auto neigh : f->neighbors) {
                table[f]->neighbors.push_back(table[neigh]);
                if (visited.find(neigh) != visited.end()) {
                    continue;
                }
                que.push(neigh);
                visited.insert(neigh);
            }
        }

    }
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值