LeetCode 138——複製帶隨機指標的連結串列
阿新 • • 發佈:2018-12-19
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」!