题目描述
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
解题思路
首先遍历结点同时,复制结点插入原来的结点的后面。然后再遍历一遍,根据原来的结点的指针关系,设置特殊指针。如果原来一个结点 a 的特殊指针指向 b,那么其复制结点 a' 指向 b 的 next,也即 b'。遍历完成的,将复制的结点从原来的链表中分离出来。
实现
/*链表结点的定义*/
public class RandomListNode {
int label;
RandomListNode next = null;
RandomListNode random = null;
RandomListNode(int label) {
this.label = label;
}
}
/*实现*/
public class Solution {
public RandomListNode Clone(RandomListNode pHead)
{
//copy
RandomListNode p = pHead;
while (p!=null){
RandomListNode node = new RandomListNode(p.label);
node.next = p.next;
p.next = node;
p = node.next;
}
//adjust special pointer
p = pHead;
while (p != null){
RandomListNode node = p.next;
if (p.random != null) node.random = p.random.next;
p = node.next;
}
//seprate from the list
RandomListNode newHead = null;
p = pHead;
RandomListNode q = newHead;
while (p != null){
if (newHead == null) newHead = q = p.next;
p.next = q.next;
p = p.next;
if (p != null) q.next = p.next;
q = q.next;
}
return newHead;
}
}