链表–复制含有随机指针节点的链表
这题如果不要求额外空间复杂度,用哈希表也可以实现,先看下用哈希表怎么做,当然,这里只是展示怎么做,但哈希表做法是不符合题目要求的哈!!
整体思路:
创建一个哈希表,用于存储原节点和对应的新节点的映射。第一次遍历链表的时候,创建新节点,并将原节点和新节点的映射存在哈希表中,设置新节点的next和rand指针,最后返回新链表的头节点
代码实现如下:
public static Node copyListWithRand(Node head){
if(head==null){
return null;
}
//创建一个哈希表,用于存储原节点和对应的新节点的映射
HashMap<Node,Node> map = new HashMap<>();
Node cur = head;
//第一次遍历,创建新节点,并将原节点和新节点的映射存在哈希表中
while(cur != null){
map.put(cur,new Node(cur.value));
cur = cur.next;
}
//重置cur指针,指向链表头部
cur = head;
//第二次遍历链表,设置新节点的next和rand指针
while(cur!=null){
//给新节点设置next指针
map.get(cur).next = map.get(cur.next);
//给新节点设置rand指针
map.get(cur).rand = map.get(cur.rand);
cur = cur.next;
}
//最后返回新链表的头结点
return map.get(head);
}
接下来,就进入正题,不借助额外空间复杂度来用代码实现题目需求
整体思路如下:
先遍历链表,复制每个结点,并把复制的节点插到原节点后面,然后设置复制节点的rand指针,最后拆分链表,
将复制链表从原链表中分离出来
代码实现如下:
public static Node copyListWithRand(Node head){
if(head==null){
return null;
}
Node cur = head;
Node next = null;
//复制每个节点
while(cur!=null){
next = cur.next;
cur.next = new Node(cur.value);
cur.next.next = next;
cur = next;
}
//重置链表头节点
cur = head;
Node curCopy = null;
//设置复制节点的rand指针
while(cur != null){
//保存下一个原节点
next = cur.next.next;
curCopy = cur.next;
curCopy.rand = cur.rand != null? cur.rand.next:null;
cur = next;
}
//拆分链表,将复制链表从原链表中分离出来
Node copyHead = head.next;
cur = head;
while(cur != null){
next = cur.next.next;
curCopy = cur.next;
cur.next = next;
curCopy.next = next != null ? next.next:null;
cur = next;
}
//返回复制链表的头节点
return copyHead;
}
感谢看到这儿的小伙伴,欢迎各位小伙伴指出文章的不当之处,那我们下篇文章再见啦