LeetCode 138. 复制带随机指针的链表

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

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

 

示例:

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

解释:
节点 1 的值是 1,它的下一个指针和随机指针都指向节点 2 。
节点 2 的值是 2,它的下一个指针指向 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 NULL;
        auto p=head;
        for(p=head;p;){
            auto q=p->next;
            auto nx=new Node(p->val,NULL,NULL);
            p->next=nx;
            nx->next=q;
            p=q;
        }
        for(p=head;p;p=p->next->next){
            if(p->random)
                p->next->random=p->random->next;
        }
        auto l=new Node(-1,NULL,NULL);
        auto cur=l;
        for(p=head;p;p=p->next){
            cur->next=p->next;
            cur=cur->next;
            p->next=p->next->next;
        }
        return l->next;
    }
};

 

posted @ 2019-07-09 23:40  YF-1994  阅读(362)  评论(0编辑  收藏  举报