快照数组

/*
 * 思路：
 * 使用一个ArrayList来存储每次快照的数组状态。
 * 使用一个HashMap来存储当前的数组状态。
 * set方法直接更新当前的数组状态。
 * snap方法将当前状态存入ArrayList并返回快照ID。
 * get方法根据快照ID返回对应状态的值。
 * 使用Java Stream API来实现这些操作。
 */
import java.util.*;
import java.util.stream.*;

class SnapshotArray {
    private List<Map<Integer, Integer>> snapshots;
    private Map<Integer, Integer> current;
    private int snapId;

    public SnapshotArray(int length) {
        snapshots = new ArrayList<>();
        current = new HashMap<>();
        snapId = 0;
    }

    public void set(int index, int val) {
        current.put(index, val);
    }

    public int snap() {
        snapshots.add(current.entrySet().stream()
            .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue)));
        return snapId++;
    }

    public int get(int index, int snap_id) {
        return snapshots.get(snap_id).getOrDefault(index, 0);
    }
}