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

Posted 2022-03-31 programyang

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

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

 

示例：

输入：
"$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;
    
;