LeetCode: 1146. 快照数组¶
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
2、解题思路¶
- 创建默认字典保存响应的值,每次打快照以后,增加一个新的快照
from collections import defaultdict from copy import copy class SnapshotArray: def __init__(self, length: int): self.length = length self.data = [defaultdict(int)] self.snap_times = 0 def set(self, index: int, val: int) -> None: if index < self.length: self.data[-1][index] = val def snap(self) -> int: self.data.append(copy(self.data[-1])) self.snap_times += 1 return self.snap_times - 1 def get(self, index: int, snap_id: int) -> int: if snap_id < self.snap_times: return self.data[snap_id][index] else: return 0 # Your SnapshotArray object will be instantiated and called as such: # obj = SnapshotArray(length) # obj.set(index,val) # param_2 = obj.snap() # param_3 = obj.get(index,snap_id)