复杂链表的拷贝
2019-07-03 本文已影响0人
小明今晚加班
题目描述
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
思路:本题是对一个复杂链表进行深拷贝,关键是结点中有random指针,如果按照正常思维(有些类似于深度遍历)去拷贝的话,复杂度太高,为O(n^2)。因此这里我们这里的拷贝分两步,第一步是拷贝next结点;第二步再考虑random指针结点。我们使用hashMap来存储每一个结点,避免random指针拷贝的时候出现重复拷贝的现象。
具体代码实现如下:
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;
//p结点指向原始链表
RandomListNode p = pHead;
RandomListNode head = new RandomListNode(-1);
//q结点用来操作新链表
RandomListNode q = head;
HashMap<RandomListNode,RandomListNode> hash = new HashMap<>();
//第一步:只拷贝next结点
while(p!=null){
RandomListNode tmp = new RandomListNode(p.label);
hash.put(p, tmp);
q.next = tmp;
q = q.next;
p = p.next;
}
//第二步:考虑random结点
p = pHead;
while(p!=null){
if(p.random!=null){
RandomListNode value = hash.get(p);
RandomListNode random = hash.get(p.random);
value.random = random;
}
p = p.next;
}
return head.next;
}
}