给定一个链表,链表里的每个节点都存在next指针外,还存在一个随机指向其他节点的指针,如何复制一个这样的链表,我觉得这是自己理解最透彻的一个算法,句句代码都有注释,欢迎大家评论不对的地方
/**
* 复制复杂链表,传入的Node为链表的头结点
* @param head
* @return
*/
public Node copyRandomList (Node head) {
Node cur = head;
// 复制各节点,并将各节点拼接
while (cur != null) {
Node temp = new Node(cur.val);
// temp的下一个指向为cur的下一个指向
temp.next = cur.next;
// cur的下一个指向为temp;
cur.next = temp;
// 开始复制下一个节点。
cur = temp.next;
}
// 开始复制random
// 先把临时Node还原为head;
// 此时数据结构为 1->1copy->2->2copy
cur = head;
while (cur!= null) {
if (cur.random != null) {
// cur的下个节点的随机指针等于cur的随机节点的下个指针
cur.next.random = cur.random.next;
}
// 每进行一次操作,cur变为下下个节点
cur = cur.next.next;
}
// 到此为止复制链表已经完成,需要拆开成两个链表
cur = head.next;
Node pre = head;
// 使用一个新的Node来记录复制node的头结点
Node res = cur;
while (cur.next != null) {
// 原节点下一个节点应该是复制后链表中的下下个节点。
pre.next = pre.next.next;
// 原节点的复制节点的下个节点应该是他的下下个节点
cur.next = cur.next.next;
pre = pre.next;
cur = cur.next;
}
// 单独处理原链表末的节点,结束while循环的时候复制链表的尾节点已经是null,所以不需要处理
pre.next = null;
return res;
}