leetcode 138: Copy List with Random Pointer

最新推荐文章于 2024-04-03 15:29:39 发布

原创最新推荐文章于 2024-04-03 15:29:39 发布 · 296 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

201 篇文章

订阅专栏

本文介绍了一种高效克隆带有随机指针的单链表方法。通过使用无序映射表来关联原始节点与复制节点的地址，实现链表的深度复制。这种方法可以避免复杂的内存管理和重复创建节点的问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Same idea with the Clone Graph. Here I simplify the unordered map to connect two pointer addresses.

/**
 * Definition for singly-linked list with a random pointer.
 * struct RandomListNode {
 *     int label;
 *     RandomListNode *next, *random;
 *     RandomListNode(int x) : label(x), next(NULL), random(NULL) {}
 * };
 */
class Solution {
public:
    RandomListNode *copyRandomList(RandomListNode *head) {
        if(!head)
            return NULL;
        unordered_map<RandomListNode*,RandomListNode*> link;
        RandomListNode* phead=new RandomListNode(head->label);
        link.insert(make_pair(head,phead));
        copy(head,phead,link);
        return phead;
    }
    void copy(RandomListNode* pold,RandomListNode* pnew,unordered_map<RandomListNode*,RandomListNode*>& link)
    {
        if(!pold)
            return;
        RandomListNode* p=pold->next;
        if(p)
        {
            if(link.find(p)!=link.end())
                pnew->next=link[p];
            else
            {
                pnew->next=new RandomListNode(p->label);
                link.insert(make_pair(p,pnew->next));
            }
        }
        p=pold->random;
        if(p)
        {
            if(link.find(p)!=link.end())
                pnew->random=link[p];
            else
            {
                pnew->random=new RandomListNode(p->label);
                link.insert(make_pair(p,pnew->random));
            }
        }
        copy(pold->next,pnew->next,link);
    }
};