【题目】 一种特殊的链表节点类描述如下: public class Node { public int value; public Node next; public Node rand; public Node(int data) { this.value = data; } } Node类中的value是节点值,next指针和正常单链表中next指针的意义 一 样,都指向下一个节点,rand指针是Node类中新增的指针,这个指 针可 能指向链表中的任意一个节点,也可能指向null。 给定一个由 Node节点类型组成的无环单链表的头节点head,请实现一个 函数完成 这个链表中所有结构的复制,并返回复制的新链表的头节点。 进阶: 不使用额外的数据结构,只用有限几个变量,且在时间复杂度为 O(N) 内完成原问题要实现的函数。
思想:这里有两种方法。第一种:使用hashmap来存放每个节点的复制节点(1,1’)。1’是节点1的复制节点。依次把每个节点和节点的复制节点都放到map里面,然后使用map来得到要拷贝的节点
代码:
public static class Node {
public int value;
public Node next;
public Node rand;
public Node(int data) {
this.value = data;
}
}
public static Node copyListWithRand1(Node head) {
HashMap<Node, Node> map = new HashMap<Node, Node>();//使用map来存放(节点,节点的复制节点)
Node cur = head;
while (cur != null) {
map.put(cur, new Node(cur.value)); //复制一开始的链表的节点
cur = cur.next;
}
cur = head;
while (cur != null) {
map.get(cur).next = map.get(cur.next); //复制节点对应相应的复制节点
map.get(cur).rand = map.get(cur.rand);
cur = cur.next;
}
return map.get(head);//返回复制的那个链表
}
第二种方法:不使用哈希表,把复制节点和节点依次连接起来,每次取两个节点,假设1的random是3,那么1’的random可以通过指向3的next得到3’;依次这样,最后把链表分隔开。
代码:
public static Node copyListWithRand2(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 res = head.next;
cur = head;
// 分隔,变成1-2-3,1'-2'-3'
while (cur != null) {
next = cur.next.next;
curCopy = cur.next;
cur.next = next;
curCopy.next = next != null ? next.next : null;
cur = next;
}
return res;
}