1. 程式人生 > >LeetCode 138——複製帶隨機指標的連結串列

LeetCode 138——複製帶隨機指標的連結串列

1. 題目

2. 解答

第一次遍歷連結串列的時候,複製舊連結串列的節點值建立一個新的連結串列,同時定義一個 unordered_map 作為雜湊表,雜湊表的鍵為舊連結串列的節點指標,值為新連結串列的節點指標。

然後,第二次遍歷連結串列,訪問舊連結串列節點的隨機指標,然後以此為鍵從 map 中取出對應的新連結串列節點指標,這也就是當前新連結串列節點的隨機指標。

/**
 * Definition for singly-linked list with a random pointer.
 * struct RandomListNode {
 *     int label;
 *     RandomListNode *next, *random;
 *     RandomListNode(int x) : label(x), next(NULL), random(NULL) {}
 * };
 */
class Solution {
public:
    RandomListNode *copyRandomList(RandomListNode *head) {
        
        unordered_map<RandomListNode *, RandomListNode *> nodemap;
        RandomListNode *temp = head;
        RandomListNode *new_head = new RandomListNode(0); //哨兵節點,方便操作
        RandomListNode *copy_temp = new_head;
 
        // 建立新連結串列
        while (temp)
        {
            copy_temp->next = new RandomListNode(temp->label);
            nodemap[temp] = copy_temp->next;
            
            temp = temp->next;
            copy_temp = copy_temp->next;
        }
        
        RandomListNode *random_temp = NULL;
        temp = head;
        copy_temp = new_head->next;
        // 填充新連結串列的隨機指標
        while (temp)
        {
            random_temp = temp->random;
            if (random_temp != NULL)   copy_temp->random = nodemap[random_temp];
            else    
                copy_temp->random = NULL;
            
            temp = temp->next;
            copy_temp = copy_temp->next;
        }
        
        return new_head->next;
    }
};

獲取更多精彩,請關注「seniusen」!