题目描述
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
https://www.nowcoder.com/practice/f836b2c43afc4b35ad6adc41ec941dba
解题想法
比较有技巧,放一张图就知道怎么做了
分为 3 步:
插入复制节点
复制 random 指向
将两个序列进行分割
代码实现
/*
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)
{
if (pHead == null){
return pHead;
}
// 1. 对每个节点进行复制,链接到原节点后面
RandomListNode head1 = pHead;
while (head1 != null){
RandomListNode temp = new RandomListNode(head1.label);
temp.next = head1.next;
head1.next = temp;
head1 = temp.next;
}
// 2. 复制random指针,
RandomListNode head2 = pHead;
while (head2 != null){
RandomListNode temp = head2.random;
if (temp != null){
head2.next.random = temp.next;
}else{
head2.next.random = null;
}
head2 = head2.next.next;
}
// 3. 将复制的节点进行拆分出来
RandomListNode head3 = pHead;
RandomListNode head4 = pHead.next;
while (head3 != null){
RandomListNode temp = head3.next;
head3.next = temp.next;
if (temp.next != null){
temp.next = temp.next.next;
}else{
temp.next = null;
}
head3 = head3.next;
}
return head4;
}
}