题目描述
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
思路:
- 遍历链表,复制每个结点,如复制结点A得到A1,将结点A1插到结点A后面;
- 重新遍历链表,复制老结点的随机指针给新结点,如A1.random = A.random.next;
- 拆分链表,将链表拆分为原链表和复制后的链表
代码如下:
/*
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 null;
}
RandomListNode p = pHead;
while (p != null) {
RandomListNode q = new RandomListNode(p.label);
q.next = p.next;
p.next = q;
p = q.next;
}
p = pHead;
while (p != null) {
RandomListNode q = p.next;
if (p.random != null) {
q.random = p.random.next;
} else {
q.random = null;
}
p = q.next;
}
p = pHead;
RandomListNode newListNode = p.next;
while (p != null) {
RandomListNode q = p.next;
p.next = q.next;
if (q.next != null) {
q.next = q.next.next;
} else {
q.next = null;
}
p = p.next;
}
return newListNode;
}
}