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」!