码迷,mamicode.com
首页 > 其他好文 > 详细

LeetCode 138. Copy List with Random Pointer

时间:2019-09-25 12:43:48      阅读:76      评论:0      收藏:0      [点我收藏+]

标签:div   问题   指针   The   build   htable   mapping   bsp   color   

本题抽象一下就是如何deep copy图的问题。由于random指针的存在,导致我们按顺序copy的时候,copy的random指针指向的node可能还没有生成。如何解决这个问题是本题的关键。

Recursive

如果递归来做,上述问题很好解决,没有生成的节点递归生成即可。

但是我们需要用一个hashtable来记录原节点与对应copy节点的映射关系,这是因为当前需要copy的节点很可能在之前已经被创建好了。这时我们只要返回已经创建好的节点的指针即可。

class Solution {
public:
    unordered_map<Node *,Node *> m; // original node -> corresponding copied node
    
    Node* copyRandomList(Node* head) {
        if (head==NULL) return NULL;
        if (m.count(head)) return m[head];
        Node *copy=new Node(head->val,NULL,NULL);
        m[head] = copy;
        copy->next = copyRandomList(head->next);
        copy->random = copyRandomList(head->random);
        return copy;
    }
};

时间复杂度 O(n) 空间复杂度 O(n)

 

Iterative (Hashtable)

递归来写就得注意random指针了。我们可以分两步来做,首先copy链表,忽视random指针,同时建立原节点和对应copy节点的映射关系。第二步根据之前的hashtable,更新正确的random指针。

*/
class Solution {
public:
    unordered_map<Node *,Node *> m; // original node -> corresponding copied node
    
    Node* copyRandomList(Node* head) {
        if (head==NULL) return NULL;
        
        // copy all the nodes (without random) and build the mapping
        Node *dummy=new Node(0,NULL,NULL), *p=dummy, *q=head;
        while (q){
            Node *copy = new Node(q->val,NULL,q->random);
            m[q] = copy;
            p->next = copy;
            p = p->next;
            q = q->next;
        }
        
        // link random
        p = dummy->next; 
        while (p){
            if (p->random!=NULL) 
                p->random = m[p->random];
            p = p->next;
        }
        
        return dummy->next;
    }
};

时间复杂度 O(n) 空间复杂度 O(n)

 

Iterative with O(1) Space

 To be continued ...

LeetCode 138. Copy List with Random Pointer

标签:div   问题   指针   The   build   htable   mapping   bsp   color   

原文地址:https://www.cnblogs.com/hankunyan/p/11583728.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!