* Clone Graph

来源:互联网 发布:千牛软件何用 编辑:程序博客网 时间:2024/06/16 04:05
class Solution {  
public:  
    UndirectedGraphNode *cloneGraph(UndirectedGraphNode *node) {  
        
        if(node==NULL) return NULL;
        map<UndirectedGraphNode*,UndirectedGraphNode*>copied;
        queue<UndirectedGraphNode *>q;
        q.push(node);
        copied[node]=new UndirectedGraphNode(node->label);
        while(!q.empty())
        {
            UndirectedGraphNode *cur=q.front();
            q.pop();
            for(int i=0;i<cur->neighbors.size();i++)
            {
                if(copied.find(cur->neighbors[i])!=copied.end())
                copied[cur]->neighbors.push_back(copied[cur->neighbors[i]]);  //copied[cur]
                else
                {
                   UndirectedGraphNode *newnode=new UndirectedGraphNode (cur->neighbors[i]->label);
                   copied[cur->neighbors[i]]=newnode;
                 copied[cur]->neighbors.push_back(newnode);
                 q.push(cur->neighbors[i]);
                }
                
            }
            
        }
        return copied[node];
    }  
}; 
0 0
原创粉丝点击