题目:
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针random指向一个随机节点),请对此
链表进行深拷贝,并返回拷贝后的头结点。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)。
解法:哈希表
public RandomListNode Clone(RandomListNode pHead) {
if (pHead == null) {
return null;
}
RandomListNode result = new RandomListNode(0);
RandomListNode pre = result;
Map<RandomListNode, RandomListNode> map = new HashMap<>();
while (pHead != null) {
RandomListNode clone = new RandomListNode(pHead.label);
pre.next = clone;
map.put(pHead, clone);
pHead = pHead.next;
pre = pre.next;
}
for (Map.Entry<RandomListNode, RandomListNode> entry : map.entrySet()) {
if (entry.getKey().random == null) {
entry.getValue().random = null;
} else {
entry.getValue().random = map.get(entry.getKey().random);
}
}
return result.next;
}
本题难点在于存在一个随机节点,所以一次遍历肯定完成不了,因为在第一次遍历的时候随机节点可能还没构造出来。
考虑使用哈希表来做,以当前节点为key,以新构造的节点为value。第一次遍历完后,可以得到一个只有后继节点没有随机节点的新链表。
那么后继节点怎么得到呢?通过遍历哈希表便可以得到:entry.getValue().random = map.get(entry.getKey().random);此行代码是整个步骤中最关键的一步。
我们来分析一下:哈希表的key为老节点,value为新节点,待做的操作是给新节点指定随机节点,新节点的随机节点一定也是新节点,而不是老节点。通过key可以得到老节点的随机节点,老节点的随机节点一定是老节点,即也是哈希表的key,通过老节点的随机节点又可以定位到新节点,那么此节点就是随机节点。
妙啊!真是太妙了!
复杂度分析:
时间复杂度:O(N),遍历链表一次O(N)
空间复杂度:O(N),哈希表存储链表节点
总结:
涉及数据结构:链表、哈希表
涉及算法:迭代