劍指offer 35 複雜鏈表的複製
題目描述
輸入一個複雜鏈表(每個節點中有節點值,以及兩個指針,一個指向下一個節點,另一個特殊指針random指向一個隨機節點),請對此鏈表進行深拷貝,並返回拷貝後的頭結點。(注意,輸出結果中請不要返回參數中的節點引用,否則判題程序會直接返回空)
分析: 即原鏈表對應的每個節點都要新建一遍,且要保證指針的指向.
方法1
原地複製,再拆分鏈表
將原鏈表的每個節點都複製一遍,且複製的節點緊隨原來的節點。全部複製完畢,再斷開與原鏈表的鏈接。
圖片來自Markus
)
import java.util.Map;
import java.util.HashMap;
/*
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 temp = p.next;
RandomListNode node = new RandomListNode(p.label);
p.next = node;
node.next = temp;
p = temp;
}
p = pHead;
// 原生鏈表的節點的指向任意節點,使複製的節點也都指向某一任意節點
while(p != null){
if(p.random != null){
RandomListNode temp = p.random;
p.next.random = temp.next;
}
p = p.next.next;
}
p = pHead;
RandomListNode newHead = p.next;
RandomListNode newnode = p.next;
// 重新連接節點,把原生節點重新連接起來,把克隆後的節點連接起來
while(p != null){
p.next = p.next.next;
if(newnode.next != null){
newnode.next = newnode.next.next;
newnode = newnode.next;
}
p = p.next;
}
return newHead;
}
}
方法二 HashMap
創建HashMap,複製結點值, 簡歷 <舊節點, 新節點> 之間的映射。
3.複製指向(next,random)
class Solution {
public Node copyRandomList(Node head) {
Map<Node, Node> map = new HashMap();
Node p = head;
// 複製新節點
while(p != null){
Node temp = new Node(p.val);
map.put(p, temp);
p = p.next;
}
// 複製指針指向。
p = head;
while(p != null){
map.get(p).next = map.get(p.next);
map.get(p).random = map.get(p.random);
p= p.next;
}
return map.get(head);
}
}