克隆图
给定无向连通图中一个节点的引用,返回该图的深拷贝(克隆)。图中的每个节点都包含它的值 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 。
思路+代码+注释:
public Node cloneGraph(Node node) {
/*
思路:使用hashMap存储已经拷贝过的节点,其中key为原节点,value为拷贝节点
递归扫描节点,如果该节点在map中不存在说明没有被拷贝过,那么新建一个节点添加到map中,在递归克隆邻居节点;如果该节点在map中,说明被拷贝过直接返回
当节点为null时也直接返回null
*/
HashMap<Node,Node> nodeMap=new HashMap<>();
return cloneGraph(node,nodeMap);
}
private Node cloneGraph(Node node, HashMap<Node,Node> map)
{
if (node==null)
{
return null;
}
if (map.containsKey(node))
{
return map.get(node);
}
Node newNode=new Node();
newNode.val=node.val;
map.put(node,newNode);
List<Node> newList=new ArrayList<>();
List<Node> nodeList=node.neighbors;
for (int i = 0; i < nodeList.size(); i++) {
Node oldNode=nodeList.get(i);
newList.add(cloneGraph(oldNode,map));
}
newNode.neighbors=newList;
return newNode;
}