美文网首页程序员
LeetCode:复制带随机指针的链表

LeetCode:复制带随机指针的链表

作者: 李海游 | 来源:发表于2020-04-07 14:02 被阅读0次

    面试题35. 复杂链表的复制

    请实现 copyRandomList 函数,复制一个复杂链表。在复杂链表中,每个节点除了有一个 next 指针指向下一个节点,还有一个 random 指针指向链表中的任意节点或者 null

    思路:
    1)原始1->2->3->N
    2)复制成为 1->1->2->2->3->3`->N
    3)在将随机指针进行复制
    4)将链表拆分成原链表和复制的链表

    class Solution {
    public:
        Node* copyRandomList(Node* head) {
            if(head==NULL)
                return NULL;
    
            /*Node* org = head;
            Node *copy = NULL;
            while (org != NULL)
            {
                copy = org;
                copy->next = org->next;
                org->next = copy;
                org = org->next->next;
            }*/
            //复制成 长链表
            Node *org = head;
            Node *pnext = NULL;
            while (org != NULL)
            {
                pnext = org->next;
                Node* tmp = new Node(org->val);
                tmp->next = pnext;
                org->next = tmp;
                org = pnext;
            }
            // 复制随机指针
            org = head;
            Node *copy = head->next;
            while (org != NULL)
            {
                //如果随即指针为空 需要判断
                copy->random = org->random? org->random->next :NULL;
                org = copy->next;
                copy = org?org->next:org;
            }
            //拆分长链表
            org = head;
            copy = head->next;
            Node *phead = copy;
            while (org != NULL)
            {
                org->next = copy->next;
                org = org->next;
                // 最后一个结点 org为空  copu->next也为空
                copy->next = org?org->next:org;
                copy = copy->next;
            }
            return phead;
        }
    };
    

    注意:代码块中注释的代码。看似也是复制成长链表,但是要记住指针不是结点,copy = org; 这种操作其实是 copy和org都指向了同一个Node,也就是存储的值都是一样的,是Node的地址,而不是copy新建了一个和org指向的结点一样的新节点。

    所以copy->next = org->next; 其实是copy->next == org->next; !!!

    相关文章

      网友评论

        本文标题:LeetCode:复制带随机指针的链表

        本文链接:https://www.haomeiwen.com/subject/caagphtx.html