leetcode 133. Clone Graph

题目概述

解题思路

这道题的主要难点在于两个：

1. 建图
2. 遍历

这道题依然采用了邻接链表的方式建图，之前我建图采用的方式是：开一个两个维度的数组，第一个维度表示节点的数目，第二个维度表示和每个节点相连的边：

class Edges
{
public:
    int weight;
    string val;
    Edges(int wi, string vi): weight(wi), val(vi){}
};

map<string, int> str2idx;
vector<Edges*> Node[N];

这种表示的缺点在于，每个节点的值不方便直接获取，因为它们都被存放在Edges这个类中。

Leetcode这种建图的方式在访问节点值的时候就更为方便：

class Node
{
public:
    int val;
    vector<Node *>neighbors;

    Node (int vi, vector<Node*> ni)
    {
        val = vi;
        neighbors = ni;
    };
    
};

下一步是遍历图。这里我们采用BFS进行原图遍历，只需维护一个set，用来判断之前的节点是否被访问过即可。每次访问到原图的某个节点，就生成相应的一个节点，然后在克隆图中插入。这样就完成了全图克隆。

方法性能

假设原图含Ｖ个节点和Ｅ条边，方法的时空复杂度均为O(V +Ｅ)：

代码示例

class Solution {
public:
    unordered_map<Node *, Node *> old2new;
    unordered_set<Node *> old_nvis;

    Node *getCorresNode(Node *old_n)
    {
        if(old2new.find(old_n) == old2new.end())
        {
            Node *new_n = new Node(old_n->val, vector<Node*>());
            old2new[old_n] = new_n;
        }
        return old2new[old_n];
    }

    Node* cloneGraph(Node* node) 
    {
        if(node == NULL)
            return node;
        queue<Node *> cur_N;
        cur_N.push(node);
        while(!cur_N.empty())
        {
            Node *temp_n = cur_N.front();
            cur_N.pop();
            if(old_nvis.find(temp_n) == old_nvis.end())
            {
                old_nvis.insert(temp_n);
                Node *new_node = getCorresNode(temp_n);

                vector<Node *> neigh = temp_n->neighbors;
                for (vector<Node *>::iterator iter = neigh.begin(); iter != neigh.end(); ++iter)
                {
                    cur_N.push(*iter);
                    Node *tmp_neigh = getCorresNode(*iter);
                    new_node->neighbors.push_back(tmp_neigh);
                }
            }
        }

        return old2new[node];
    }
};