leetcode 138. Copy List with Random Pointer

最新推荐文章于 2025-09-11 19:30:36 发布

转载最新推荐文章于 2025-09-11 19:30:36 发布 · 57 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/ymjyqsx/p/10802932.html

文章标签：

#数据结构与算法

博客介绍了复制带随机指针链表的方法，分三步进行：先在原有链表每个节点后增加值相同的节点，接着在新节点中增加随机节点，最后分离原节点和新节点。同时提醒要注意判断null的情况，以及代码不适用于头节点为空的情况。

138. Copy List with Random Pointer

分三步：1.在原有list每个节点的后面增加与原节点值相同的节点

　　　　2.在这些新生成的节点中增加随机节点

　　　　3.将原有的节点和新生成的节点进行分离

注意：

if(cur->random)
randNode = cur->random->next;
else
randNode = NULL;

if(res->next)
res->next = res->next->next;
else
res->next = NULL;

这两个地方都要判断null的情况。还有这个代码不适合head == NULL的情况，所以一开始也要写为空的返回情况。

class Solution {
public:
    Node* copyRandomList(Node* head) {
        if(head == NULL)
            return NULL;
        Node* cur = head;
        while(cur){
            Node* tmp = cur->next;
            Node* newNode = new Node(cur->val);
            cur->next = newNode;
            newNode->next = tmp;
            cur = tmp;
        }
        cur = head;
        while(cur){
            Node* tmp = cur->next;
            Node* randNode;
            if(cur->random)
                randNode = cur->random->next;
            else
                randNode = NULL;
            tmp->random = randNode;
            cur = cur->next->next;
        }
        cur = head;
        Node* res = head->next;
        Node* result = head->next;
        while(cur){
            cur->next = cur->next->next;
            if(res->next)
                res->next = res->next->next;
            else
                res->next = NULL;
            cur = cur->next;
            res = res->next;
        }
        return result;
    }
};