输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。下面看一看时间复杂度为O(N)的复杂链表复制算法。
假如被复制的链表:
思路:分3步;
1、将复制的节点插入到对应节点的后面,形成一个新的链表。
2、复制随机节点,通过原节点找到随机节点的位置
3、拆分链表,将原链表与新链表进行拆分
最终复制成功,代码如下:
public class Solution {
public RandomListNode Clone(RandomListNode pHead)
{
if (pHead == null) {
return null;
}
//复制并且插入
RandomListNode p = pHead;
while (p != null) {
RandomListNode tmp = new RandomListNode(p.label);
tmp.next = p.next;
p.next = tmp;
p = tmp.next;
}
//复制random指针
p = pHead;
while (p != null) {
if (p.random != null) {
p.next.random = p.random.next;
}
p = p.next.next;
}
//拆分链表
RandomListNode head = pHead.next;
RandomListNode q = head;
p = pHead;
while (q.next != null) {
p.next = q.next;
p = p.next;
q.next = p.next;
q = q.next;
}
p.next = null; //最后将原来链表的尾部设置为null
return head;
}
}