思想 (链表, 模拟题)
该题的解法分为3步,如下图所示:
## 题目
请实现一个函数可以复制一个复杂链表。
在复杂链表中,每个结点除了有一个指针指向下一个结点外,还有一个额外的指针指向链表中的任意结点或者null。
注意:
函数结束后原链表要与输入时保持一致。
数据范围
链表长度 [0,500].
代码
/**
* Definition for singly-linked list with a random pointer.
* class ListNode {
* int val;
* ListNode next, random;
* ListNode(int x) { this.val = x; }
* };
*/
class Solution {
public ListNode copyRandomList(ListNode head) {
// 1. 将每一个点复制出来
for (ListNode p = head; p != null; ) {
ListNode np = new ListNode(p.val);
ListNode next = p.next;
p.next = np;
np.next = next;
p = next;
}
// System.out.println("1");
// 2. random指针的处理
for (ListNode p = head; p != null; p = p.next.next) {
if (p.random != null)
p.next.random = p.random.next;
}
// System.out.println("2");
// // 3. 将复制好的新的链表拆出来
ListNode dummy = new ListNode(-1);
ListNode cur = dummy;
for (ListNode p = head; p != null;) {
cur.next = p.next;
cur = cur.next;
p.next = p.next.next;
p = p.next;
}
// System.out.println("3");
return dummy.next;
}
}