复杂链表的复制
请实现 copyRandomList 函数,复制一个复杂链表。在复杂链表中,每个节点除了有一个 next 指针指向下一个节点,还有一个 random 指针指向链表中的任意节点或者 null。
示例 1:
输入:head = [[7,null],[13,0],[11,4],[10,2],[1,0]]
输出:[[7,null],[13,0],[11,4],[10,2],[1,0]]
示例 2:
输入:head = [[1,1],[2,1]]
输出:[[1,1],[2,1]]
示例 3:
输入:head = [[3,null],[3,0],[3,null]]
输出:[[3,null],[3,0],[3,null]]
示例 4:
输入:head = []
输出:[]
解释:给定的链表为空(空指针),因此返回 null。
提示:
-10000 <= Node.val <= 10000
Node.random 为空(null)或指向链表中的节点。
节点数目不超过 1000 。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/fu-za-lian-biao-de-fu-zhi-lcof
解题思路
1、map<Node,Node>存储节点信息;
2、复制节点:查询集合中是否存在当前节点,若存在(已复制),直接返回已复制好的节点;若不存在,复制节点;
3、递归next指向节点,由于是递归操作,copyNode.next = dfs(node.next)会完成所有节点的复制;
4、递归random指向节点,由于所有节点已复制完成,copyNode.random = dfs(node.random)会完成所有节点的random属性赋值.
代码1
/*
// Definition for a Node.
class Node {
int val;
Node next;
Node random;
public Node(int val) {
this.val = val;
this.next = null;
this.random = null;
}
}
*/
//深度优先遍历
class Solution {
Map<Node,Node> map = new HashMap<>(); //存储节点信息
public Node copyRandomList(Node head) {
return dfs(head);
}
public Node dfs(Node node){
if(node == null) return null;
Node getValue = map.get(node);
if(getValue != null){ //节点已复制
return getValue;
}
Node copyNode = new Node(node.val); //复制节点
map.put(node,copyNode); //存入集合
copyNode.next = dfs(node.next); //递归next指向节点,最终会完成整条链表的节点复制
copyNode.random = dfs(node.random); //递归random指向节点,最终完成节点random的赋值
return copyNode;
}
}