具有随机指针的链表的深层副本

时间:2019-10-01 05:47:28

标签: c++ pointers linked-list

这是一个leetcode面试问题,要求提供链表的深层副本,该链表具有一个random成员,该成员可以指向任何元素。 (https://leetcode.com/problems/copy-list-with-random-pointer)。我不知道为什么我的解决方案会因访问错误而给出运行时错误。

这是我的解决方法:

/*
// Definition for a Node.
class Node {
public:
    int val;
    Node* next;
    Node* random;

    Node() {}

    Node(int _val, Node* _next, Node* _random) {
        val = _val;
        next = _next;
        random = _random;
    }
};
*/
class Solution {
public:
    Node* copyRandomList(Node* head) {
        unordered_map<Node*, Node*> node_map;
        Node* copy_head = new Node();

        Node* node = head;
        Node* copy = copy_head;

        while (node) {
            node_map[node] = copy;
            copy->val = node->val;
            if (node->next)
                copy->next = new Node();
            node = node->next;
            copy = copy->next;
        }

        node = head;
        copy = copy_head;

        while (node) {
            copy->random = node_map[node->random];
            node = node->next;
            copy = copy->next;
        }

        return copy_head;        
    }
};

2 个答案:

答案 0 :(得分:3)

您的第一个while循环不是分配链表的新副本的首选或规范方法。在循环内(而不是循环外)应该只有1个对new的调用。如果head为空,则您仍在分配不应分配的Node

更重要的是,复制代码使用的Node的默认构造函数不会初始化任何数据成员,尤其是next指针。因此,您最后复制的节点将具有一个随机的不确定值,该值可能不会为null,因此不会正确终止列表。

尝试更多类似的方法:

/*
// Definition for a Node.
class Node {
public:
    int val;
    Node* next;
    Node* random;

    Node() {}

    Node(int _val, Node* _next, Node* _random) {
        val = _val;
        next = _next;
        random = _random;
    }
};
*/

class Solution {
public:
    Node* copyRandomList(Node* head) {
        if (!head)
            return nullptr;

        unordered_map<Node*, Node*> node_map;
        Node* copy_head = nullptr;
        Node** copy = &copy_head;
        Node* node = head;

        do {
            *copy = new Node(node->val, nullptr, nullptr);
            node_map.insert(make_pair(node, *copy));
            copy = &((*copy)->next);
            node = node->next;
        }
        while (node);

        node = copy_head;

        do {
            if (head->random)
                node->random = node_map[head->random];
            node = node->next;
            head = head->next;
        }
        while (head);

        return copy_head;
    }
};

答案 1 :(得分:1)

问题在于您的副本没有以空指针终止。也就是说,其最后一个节点不在其nullptr成员变量中存储next

您可以通过new node()创建副本的节点,该节点将调用默认的构造函数。但是,此默认构造函数不执行任何操作,包括不初始化成员变量。因此,您需要将最后一个节点的next手动设置为nullptr