LeetCode 1146. 快照陣列（雜湊map + 有序map）

1. 題目

實現支援下列介面的「快照陣列」- SnapshotArray：

• SnapshotArray(int length) - 初始化一個與指定長度相等的 類陣列 的資料結構。初始時，每個元素都等於 0。
• void set(index, val) - 會將指定索引 index 處的元素設定為 val。
• int snap() - 獲取該陣列的快照，並返回快照的編號 snap_id（快照號是呼叫 snap() 的總次數減去 1）。
• int get(index, snap_id) - 根據指定的 snap_id 選擇快照，並返回該快照指定索引 index 的值。

範例：
輸入：["SnapshotArray","set","snap","set","get"]
     [[3],[0,5],[],[0,6],[0,0]]
輸出：[null,null,0,null,5]
解釋：
SnapshotArray snapshotArr = new SnapshotArray(3); 
// 初始化一個長度為 3 的快照陣列
snapshotArr.set(0,5);  
// 令 array[0] = 5
snapshotArr.snap();  
// 獲取快照，返回 snap_id = 0
snapshotArr.set(0,6);
snapshotArr.get(0,0);  
// 獲取 snap_id = 0 的快照中 array[0] 的值，返回 5
 
提示：
1 <= length <= 50000
題目最多進行50000 次set，snap，和 get的呼叫 。
0 <= index < length
0 <= snap_id < 我們呼叫 snap() 的總次數
0 <= val <= 10^9

來源：力扣（LeetCode） 連結：https://leetcode-cn.com/problems/snapshot-array
著作權歸領釦網路所有。商業轉載請聯絡官方授權，非商業轉載請註明出處。

2. 解題

class SnapshotArray {
    int time = 0;
    unordered_map<int,map<int,int>> m;//idx, time, val
public:
    SnapshotArray(int length) { }
    
    void set(int index, int val) {
        m[index][time] = val;
    }
    
    int snap() {
        return time++;
    }
    
    int get(int index, int snap_id) {
        if(!m.count(index))
            return 0;
        auto it = m[index].upper_bound(snap_id);
        //snap_id key 時間 有序 二分查詢上限，上限之前的為答案
        if(it == m[index].begin())
            return 0;
        return (--it)->second;
    }
};

504 ms 79 MB

---

我的CSDN部落格地址 https://michael.blog.csdn.net/

長按或掃碼關注我的公眾號（Michael阿明），一起加油、一起學習進步！