leetcode第二题java_LeetCode 第 148 场周赛 第二题（5149.快照数组）JAVA 题解（TreeMap）...

实现支持下列接口的「快照数组」- 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 

0 <= snap_id

0 <= val <= 10^9

题解：

public class SnapshotArray {

List> list = new ArrayList<>();

int version = 0;

public SnapshotArray(int length) {

version = 0;

for (int i = 0; i < length; i++) {

TreeMap t = new TreeMap<>();

t.put(version, 0);

list.add(t);

}

}

public void set(int index, int val) {

list.get(index).put(version, val);

}

public int snap() {

int res = version;

version ++;

return res;

}

public int get(int index, int snap_id) {

TreeMap t = list.get(index);

return t.floorEntry(snap_id).getValue();

}

}