题目描述
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
public class RandomListNode {
int label;
RandomListNode next = null;
RandomListNode random = null;
RandomListNode(int label) {
this.label = label;
}
}
解题方法
- 遍历链表,复制每个结点,如复制结点A得到A‘,将结点A’插到结点A后面。
- 重新遍历链表,复制老结点的随机指针给新结点,如A‘.random = A.random.next。
- 拆分链表,将链表拆分为原链表和复制后的链表。
public RandomListNode Clone(RandomListNode pHead) {
if(pHead==null)
return null;
RandomListNode front = pHead;
//复制当前节点并插入到该节点后面
while(front != null) {
RandomListNode node = new RandomListNode(front.label);
node.next = front.next;
front.next = node;
front=front.next.next;
}
//复制random值
front = pHead;
while(front != null) {
if(front.random == null) //random值可能为null
front.next.random = null;
else
front.next.random = front.random.next;
front = front.next.next;
}
front = pHead;
RandomListNode newHead = pHead.next, temp = newHead;
//拆分
while(front != null) {
front.next = front.next.next;
front = front.next;
if(front == null)
temp.next = front;
else
temp.next = front.next;
temp = temp.next;
}
return newHead;
}