Leetcode133. 克隆图

题目

给定无向连通图中一个节点的引用,返回该图的深拷贝(克隆)。图中的每个节点都包含它的值 val(Int) 和其邻居的列表(list[Node])。

示例:

输入:
{"$id":"1","neighbors":[{"$id":"2","neighbors":[{"$ref":"1"},{"$id":"3","neighbors":[{"$ref":"2"},{"$id":"4","neighbors":[{"$ref":"3"},{"$ref":"1"}],"val":4}],"val":3}],"val":2},{"$ref":"4"}],"val":1}

解释:
节点 1 的值是 1,它有两个邻居:节点 2 和 4 。
节点 2 的值是 2,它有两个邻居:节点 1 和 3 。
节点 3 的值是 3,它有两个邻居:节点 2 和 4 。
节点 4 的值是 4,它有两个邻居:节点 1 和 3 。

提示:

节点数介于 1 到 100 之间。
无向图是一个简单图,这意味着图中没有重复的边,也没有自环。
由于图是无向的,如果节点 p 是节点 q 的邻居,那么节点 q 也必须是节点 p 的邻居。
必须将给定节点的拷贝作为对克隆图的引用返回。

C++解法

#include 
#include 
#include 
#include 
#include 
using namespace std;
class Node {
public:
    int val;
    vector neighbors;

    Node() {
        
    }

    Node(int _val, vector _neighbors) {
        val = _val;
        neighbors = _neighbors;
    }
};
class Solution {
public:
    map graph;
    Node * cloneGraph(Node * node) {
        if (node == NULL) return NULL;
        Node * cloned = graph[node->val];
        if (cloned) return cloned;
        cloned = new Node();
        cloned->val = node->val;
        graph[node->val] = cloned;
        for (auto neighbor: node->neighbors) {
            cloned->neighbors.push_back(cloneGraph(neighbor));
        }
        return cloned;
    }
};

int main(int argc, const char * argv[]) {
    // insert code here...
    Solution solution;
    Node node;
    node.val = 1;
    Node * node2 = new Node();
    node2->val = 2;
    Node * node4 = new Node();
    node4->val = 4;
    Node * node3 = new Node();
    node3->val = 3;
    node.neighbors.push_back(node2);
    node.neighbors.push_back(node4);
    node2->neighbors.push_back(node3);
    node4->neighbors.push_back(node3);
    auto result = solution.cloneGraph(&node);
    queue q;
    set visited;
    q.push(result);
    cout << result->val << endl;
    visited.insert(result->val);
    while (!q.empty()) {
        auto element = q.front();
        for (auto item: element->neighbors) {
            if (visited.count(item->val) == 0) {
                q.push(item);
                visited.insert(item->val);
                cout << item->val << endl;
            }
        }
        q.pop();
    }
    return 0;
}

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/clone-graph

你可能感兴趣的:(Leetcode133. 克隆图)