Leetcode 138. 复制带随机指针的链表 解题思路及C++实现

解题思路：

主要包括三步。

第一步是遍历一次链表，复制其每一个节点，并将所复制的节点接在其后。

第二步是遍历一次链表，解决拷贝节点的random指针的指向。

第三步是从这个大链表中，拆出原有链表和拷贝链表。

具体图解，课参考LeetCode官方图解。

 

/*
// 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 head;
        Node* tmp = head;
        //遍历链表，复制每一个节点，并将其放在所复制节点之后
        while(tmp){
            Node* hh = new Node(tmp->val);
            hh->next = tmp->next;
            tmp->next = hh;
            tmp = hh->next;
        }
        tmp = head;
        //遍历链表，使得复制的节点的random指针指向正确的节点
        while(tmp){
            if(tmp->random == NULL) tmp->next->random = NULL;
            else tmp->next->random = tmp->random->next;
            if(tmp->next)
                tmp = tmp->next->next;
            else tmp = NULL;
        }
        Node* res = head->next;
        // tmp 记录原始的链表节点，tmp2记录新复制的链表节点
        tmp = head;   
        Node* tmp2 = head;
        while(tmp){
            tmp2 = tmp->next;
            tmp->next = tmp2->next;
            tmp = tmp->next;
            if(tmp) tmp2->next = tmp->next;
            else tmp2->next = NULL;
        }
        return res;
    }
};