LeedCode算法题138:复制带随机指针的链表解析

给定一个链表,每个节点包含一个额外增加的随机指针,该指针可以指向链表中的任何节点或空节点。

要求返回这个链表的深拷贝。

示例:
在这里插入图片描述

输入:
{"$id":"1","next":{"$id":"2","next":null,"random":{"$ref":"2"},"val":2},"random":{"$ref":"2"},"val":1}

解释:
节点 1 的值是 1,它的下一个指针和随机指针都指向节点 2 。
节点 2 的值是 2,它的下一个指针指向 null,随机指针指向它自己。

提示:
你必须返回给定头的拷贝作为对克隆列表的引用。

这个题目和133克隆图的题目就是一样的思想,一模一样。。。。。直接设置一个哈希表存放已经遍历过的节点,键存放源列表节点位置,值存放克隆列表节点位置。深度优先搜索即可。

C++源代码:

/*
// 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*> m;
        return clone(head, m);
    }
    Node* clone(Node* node, unordered_map<Node*, Node*>& m){
        if (!node) return node;
        if (m.count(node)) return m[node];
        Node *newNode = new Node(node->val, NULL, NULL);
        m[node] = newNode;
        newNode->next = clone(node->next, m);
        newNode->random = clone(node->random, m);
        return newNode;
    }
};

python3源代码:

"""
# Definition for a Node.
class Node:
    def __init__(self, val, next, random):
        self.val = val
        self.next = next
        self.random = random
"""
class Solution:
    def copyRandomList(self, head: 'Node') -> 'Node':
        m = {}
        return self.clone(head, m)
    def clone(self, node, m):
        if node==None: return node
        if node in m.keys(): return m[node]
        newNode = Node(node.val, None, None)
        m[node] = newNode
        newNode.next = self.clone(node.next, m)
        newNode.random = self.clone(node.random, m)
        return newNode
    

猜你喜欢

转载自blog.csdn.net/x603560617/article/details/88068029