题目
请实现 copyRandomList(Node head) 函数,复制一个复杂链表。在复杂链表中,每个节点除了有一个 next 指针指向下一个节点,还有一个 random 指针指向链表中的任意节点或者 null
这题刚开始没看懂题意,大概题意就是拷贝出来一个给出的复杂链表,链表中每个节点有两个指针,指向下一个节点和指向任意节点(或者是指向null),所以在拷贝时,不仅要拷贝链表的顺序,还要把链表中每个节点指向的任意节点拷贝过去,拷贝链表的顺序很容易,关键是如何拷贝当前节点指向的任意节点。
在这里,复制的意思是指 深拷贝(Deep Copy),类似我们常用的“复制粘贴”,事实上,与此对应的还有 浅拷贝,它们的区别是:
- 浅拷贝只复制指向某个对象的指针,而不复制对象本身,新旧对象还是共享同一块内存。但深拷贝会另外创造一个一模一样的对象,新对象跟原对象不共享内存,修改新对象不会改到原对象。
方法1
第一次遍历原链表时不考虑random指针,复制原始链表上的每个节点,用next进行连接;
第二次遍历原链表时只考虑random指针,由于不能确定原始链表中当前节点random指向的节点,所以要从头节点开始遍历,找到之后再将当前节点的复制节点random指针指向找到的这个节点。
由于每次定位当前节点的random节点都需要从头进行遍历O(n),所以方法1的总时间复杂度为O(n^2)
方法2
用空间换时间,使用map数据结构,保存原节点与新节点的对应关系<oldNode, newNode>,这样在确定random指针时可快速定位新节点。
时间复杂度为O(n),空间复杂度为O(n)
方法3
第一步是将新节点先插入至对应原节点的后面
第一行是原始的链表,A的random指针指向C
第二步是将新节点的random指针指向旧节点的random指向的节点的后面那个指针,比如A’就要指向C’
第三步将新链表拆分出来
-
奇数位的节点用next链接起来就是原始链表
-
偶数位节点链接起来就是复制出来的链表
class Solution {
public Node copyRandomList(Node head) {
if (head == null) return null;
Node cur = head;
// 第一步 复制各节点到其next位置
while (cur != null) {
Node tmp = new Node(cur.val);
tmp.next = cur.next;
cur.next = tmp;
cur = tmp.next;
}
// 第二步 构建各新节点的random指向
cur = head;
while(cur != null) {
if (cur.random != null) {
cur.next.random = cur.random.next;
}
cur = cur.next.next;
}
// 第三步 拆分链表为两个链表
cur = head.next;
Node pre = head;
Node res = head.next;
while(cur.next != null) {
pre.next = pre.next.next;
cur.next = cur.next.next;
pre = pre.next;
cur = cur.next;
}
pre.next = null; // 原链表尾部设置为null
return res;
}
}
时间复杂度为O(n),空间复杂度为O(1)