LeetCode 133 Clone Graph

LeetCode 133 Clone Graph

题目的意思是，给你一个图，你给他返回一个复制的图。
所有的节点你都需要自己New 而不能用他给的参数赋值，因为那是个指针。

每个节点的label都是唯一的，由于存在环，防止无限递归，可以用map将创建的节点存起来。

c++

class Solution {
public:
    UndirectedGraphNode* ans;
    map<int,UndirectedGraphNode*> vis;
    UndirectedGraphNode *cloneGraph(UndirectedGraphNode *node) {
        if(node==NULL)
            return ans;
        ans = new UndirectedGraphNode(node->label);
        vis[ans->label] = ans;
        dfs(node,ans);
        return ans;
    }
    
    void dfs(UndirectedGraphNode *node,UndirectedGraphNode *ans)
    {
        int l=node->neighbors.size();
        for(int i=0;i<l;i++)
        {
            if(vis[node->neighbors[i]->label]==NULL)
            {    
                vis[node->neighbors[i]->label] = new UndirectedGraphNode(node->neighbors[i]->label);
                ans->neighbors.push_back( vis[node->neighbors[i]->label]);
                
                dfs(node->neighbors[i],ans->neighbors[i]);
            }
            else
                ans->neighbors.push_back(vis[node->neighbors[i]->label]);
            
        }
    }
    
    
};