题目:
实现支持下列接口的「快照数组」- 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
解答:
代码:
class SnapshotArray {
//当前快照编号,初始值为0
private int curSnapId;
//每个index的历史修改记录
private final Map<Integer,List<int[]>> history=new HashMap<>();
public SnapshotArray(int length) {
}
public void set(int index, int val) {
history.computeIfAbsent(index,k->new ArrayList<>()).add(new int[]{curSnapId,val});
}
public int snap() {
return curSnapId++;
}
public int get(int index, int snap_id) {
if(!history.containsKey(index)){
return 0;
}
List<int[]> h=history.get(index);
int j=search(h,snap_id);
return j<0?0:h.get(j)[1];
}
//返回最大的下标i,满足h[i][0]<=x
//如果不存在则返回-1
private int search(List<int[]> h,int x){
//开区间(left,right)
int left=-1;
int right=h.size();
while(left+1<right){//区间不为空
//循环不变量
// h[left][0] <= x
// h[right][1] > x
int mid=left+(right-left)/2;
if(h.get(mid)[0]<=x){
left=mid;//区间缩小为(mid,right)
}else{
right=mid;//区间缩小为(left,mid)
}
}
//根据循环不变量,此时h[left][0]<=x且h[left+1][0]=h[right][0]>x
//所以left是最大的满足h[left][0]<=x的下标
//如果不存在,则left为其初始值-1
return left;
}
}
/**
* Your SnapshotArray object will be instantiated and called as such:
* SnapshotArray obj = new SnapshotArray(length);
* obj.set(index,val);
* int param_2 = obj.snap();
* int param_3 = obj.get(index,snap_id);
*/