复制带随机链表的指针

题目要求:
给定一个链表,每个节点包含一个额外增加的随机指针,该指针可以指向链表中的任何节点或空节点。
要求返回这个链表的深拷贝。

这道题我知道的有两种解法:一种是遍历链表,找到每一个节点的随机指针,但是这样做时间复杂度是O(n^2),另外一种是拷贝节点的方法,下面我用一幅图具体讲一下节点的拷贝思路。
在这里插入图片描述
下面配以代码具体说明:

class Solution {
public:
    Node* copyRandomList(Node* head) {
        if(head==NULL)
            return head;
        //1、拷贝每一个节点到原链表中
        Node* cur=head;
        while(cur)
        {
            Node* copy=new Node(cur->val);
            Node* next=cur->next;
            cur->next=copy;
            copy->next=next;
            cur=next;
        }
        //2、在复制完的新链表中确定random指针
        cur=head;
        while(cur)
        {
            Node* copy=cur->next;
            if(cur->random)
            {
                copy->random=cur->random->next;
            }
            else
                copy->random=NULL;
            cur=copy->next;
        }
        //3、把copy的节点从原链表中拆解下来
        Node* copyhead,*tail;
        copyhead=tail=new Node;
        cur=head;
        while(cur)
        {
            Node* copy=cur->next;
            Node* next=copy->next;
            //新链表的尾插
            tail->next=copy;
            tail=copy;
            cur->next=next;
            cur=next;
        }
        Node* newhead=copyhead->next;
        delete copyhead;
        return newhead;
    }
};

这道题需要注意的是返回的是新开辟的头节点copyhead的下一个位置,最后再释放开辟的头节点copyhead。

猜你喜欢

转载自blog.csdn.net/yam_sunshine/article/details/88710484