复杂链表的复制
题目
请实现 copyRandomList 函数,复制一个复杂链表。在复杂链表中,每个节点除了有一个 next 指针指向下一个节点,还有一个 random 指针指向链表中的任意节点或者 null。
思路
开始我个人想法是直接新建一个链表,然后新建所有的节点,再给所有新节点设计next和random指向,完成一个复杂链表的复制。 但是,当我自己实现起来的时候发现太复杂了,有点难… 好在有讨论区,看了看各位大神的解法,个人觉得以拼接+分割的方式比较简单
过程
首先,如下图我们创建一个复杂链表 (这里为了方便画图,就只给一个节点设置了random)
在原有链表上复制节点本身,生成一个新的链表,像下图这样
我用黄色标记的节点就是新建立的节点。然后我们在给新节点设计好random指向就完成了拼接操作 (以下为设计random指向的代码)
cur = head;
while (cur != null) {
if (cur.random != null) {
cur.next.random = cur.random.next;
}
cur = cur.next.next;
}
最后,我们再进行分割就大功告成啦!
cur = head.next;
Node pre = head;
Node res = head.next;
while (cur.next != null) {
pre.next = pre.next.next;
cur.next = cur.next.next;
pre = pre.next;
cur = cur.next;
}
最重要的还是要有思路,思路有了,代码自然也就迎刃而解了。
完整实现代码
class Node {
int val;
Node next;
Node random;
public Node(int val) {
this.val = val;
this.next = null;
this.random = null;
}
}
public Node copy(Node head) {
if (head == null) {
return null;
}
Node cur = head;
while (cur != null) {
Node tmp = new Node(cur.val);
tmp.next = cur.next;
cur.next = tmp;
cur = tmp.next;
}
cur = head;
while (cur != null) {
if (cur.random != null) {
cur.next.random = cur.random.next;
}
cur = cur.next.next;
}
cur = head.next;
Node pre = head;
Node res = head.next;
while (cur.next != null) {
pre.next = pre.next.next;
cur.next = cur.next.next;
pre = pre.next;
cur = cur.next;
}
pre.next = null;
return res;
}
|