剑指 Offer 35. 复杂链表的复制 LCOF

仅供自己学习

 

思路：

之前也做过复制链表的题，当时也是用的hash表将原链表作为key，将复制链表作为value一一映射，当所有的链表都存放在hash表后，就能很快速的找到下一个节点和random指向的节点。

为什么不直接复制后直接赋值，如果这样直接赋值，指向的不是复制的而是原链表的节点。如果遍历的时候还复制该节点指向的random的节点来给复制节点，那么到random这个节点赋值的时候，指向的又不是同一个了。

代码：

/*
// Definition for a Node.
class Node {
public:
    int val;
    Node* next;
    Node* random;
    
    Node(int _val) {
        val = _val;
        next = NULL;
        random = NULL;
    }
};
*/
class Solution {
public:
    Node* copyRandomList(Node* head) {
        if(head==NULL) return NULL;
        Node* cur=head;
        unordered_map<Node*,Node*> mp;
        while(cur!=NULL){
            mp[cur]=new Node(cur->val);
            cur=cur->next;
        }
        cur=head;
        while(cur!=NULL){
            mp[cur]->next=mp[cur->next];
            mp[cur]->random=mp[cur->random];
            cur=cur->next;
        }
        return mp[head];
    }
};

 

还有一种是在原链表操作的方法，拼接和拆分。我们遍历原链表，在每个原链表节点添加复制节点，并将原链表节点指向这复制节点，当完成后再将复制节点的random指向，也能解决random指向的问题。当我们拆分的时候要将原链表还原，否则会报错。拆分的时候就是将原链表和复制链表分离出来，用一个指针指向复制链表的头节点，然后再用一个cur指针开始分离复制链表，用一个pre指针复原原链表。

 

代码：

/*
// Definition for a Node.
class Node {
public:
    int val;
    Node* next;
    Node* random;
    
    Node(int _val) {
        val = _val;
        next = NULL;
        random = NULL;
    }
};
*/
class Solution {
public:
    Node* copyRandomList(Node* head) {
        if(head==NULL) return NULL;
        Node* cur=head;
        while(cur!=NULL){
            Node* temp=new Node(cur->val);
            temp->next=cur->next;
            cur->next=temp;
            cur= temp->next;
        }
        cur=head;
        while(cur!=NULL){
            if(cur->random!=NULL)
                cur->next->random=cur->random->next;
            cur=cur->next->next;
        }
        Node* res=head->next;
        cur=head->next;
        Node* pre=head;
        while(cur->next!=NULL){
            pre->next=pre->next->next;
            cur->next=cur->next->next;
            pre=pre->next;
            cur=cur->next;
        }
        pre->next=NULL;
        return res;
    }
};