剑指offer-复杂链表的复制C++实现
#include <iostream>
#include <unordered_map>
using namespace std;
struct RandomListNode {
int label;
struct RandomListNode *next, *random;
RandomListNode(int x) : label(x), next(nullptr), random(nullptr) {}
};
class Solution {
public:
RandomListNode *Clone(RandomListNode *head) {
if (head == nullptr) {
return nullptr;
}
// 当前结点没复制时,递归查找存储新结点的next结点和random结点
if (!hashtable.count(head)) {
// 构造当前结点的副本
RandomListNode *copy_head = new RandomListNode(head->label);
// 利用哈希表存储新构造的结点
hashtable[head] = copy_head;
// 递归连接新结点的后继结点
copy_head->next = Clone(head->next);
// 递归查找新结点的random结点,若已存在于哈希表中,则返回哈希表中key对应的value值(结点)
copy_head->random = Clone(head->random);
}
// (自底向上)每次递归结束返回构造好的新结点
return hashtable[head];
}
private:
unordered_map<RandomListNode *, RandomListNode *> hashtable;
};
思路:
这是一道中等难度的链表题,总体思路是回溯+哈希表,使用哈希表存储新构建的结点,若结点存在于表中,则根据key值进行获取该结点,避免重复查找(记忆化搜索),根据回溯算法,自底向上递归查找新构建结点的next结点和random结点。