[leetcode]133. Clone Graph 克隆图

 

题目

给定一个无向图的节点，克隆能克隆的一切

 

思路

1--2
|
3--5

以上图为例，

node    neighbor

  1         2， 3

  2         1 

  3         1， 5

  5         3

首先从1开始,  将（node，newNode）put到HashMap中

node        newNode

  1                1 

然后遍历该node的所有neighbor

node    neighbor

  1         2， 3

此时遍历到2 

将（node，newNode）put到HashMap中

node        newNode

  1                1 -- 2  // newNode.neighbors.add(clone)

  2               2 

 

代码

public class Solution {
    
    Map<UndirectedGraphNode, UndirectedGraphNode> map = new HashMap<>();
    
    public UndirectedGraphNode cloneGraph(UndirectedGraphNode node) {
        if (node == null) {
            return null;
        }
        
        //DFS
        if (map.containsKey(node)) {
            return map.get(node);
        }
        
        UndirectedGraphNode newNode = new UndirectedGraphNode(node.label);
        map.put(node, newNode);
        for (UndirectedGraphNode nei : node.neighbors) {
            UndirectedGraphNode clone = cloneGraph(nei);
            newNode.neighbors.add(clone);
        }
        return newNode;
    }       
}