LeetCode 每日一题 981. 基于时间的键值存储

981. 基于时间的键值存储

创建一个基于时间的键值存储类 TimeMap，它支持下面两个操作：

1.set(string key, string value, int timestamp)

• 存储键 key、值 value，以及给定的时间戳 timestamp。

2.get(string key, int timestamp)

• 返回先前调用 set(key, value, timestamp_prev) 所存储的值，其中 timestamp_prev <= timestamp。
• 如果有多个这样的值，则返回对应最大的 timestamp_prev 的那个值。
• 如果没有值，则返回空字符串（""）。

示例 1：

输入：inputs = ["TimeMap","set","get","get","set","get","get"], inputs = [[],["foo","bar",1],["foo",1],["foo",3],["foo","bar2",4],["foo",4],["foo",5]]
输出：[null,null,"bar","bar",null,"bar2","bar2"]
解释：  
TimeMap kv;   
kv.set("foo", "bar", 1); // 存储键 "foo" 和值 "bar" 以及时间戳 timestamp = 1   
kv.get("foo", 1);  // 输出 "bar"   
kv.get("foo", 3); // 输出 "bar" 因为在时间戳 3 和时间戳 2 处没有对应 "foo" 的值，所以唯一的值位于时间戳 1 处（即 "bar"）   
kv.set("foo", "bar2", 4);   
kv.get("foo", 4); // 输出 "bar2"   
kv.get("foo", 5); // 输出 "bar2"   

示例 2：

输入：inputs = ["TimeMap","set","set","get","get","get","get","get"], inputs = [[],["love","high",10],["love","low",20],["love",5],["love",10],["love",15],["love",20],["love",25]]
输出：[null,null,null,"","high","high","low","low"]

提示：

• 所有的键/值字符串都是小写的。
• 所有的键/值字符串长度都在 [1, 100] 范围内。
• 所有 TimeMap.set 操作中的时间戳 timestamps 都是严格递增的。
• 1 <= timestamp <= 10^7
• TimeMap.set 和 TimeMap.get 函数在每个测试用例中将（组合）调用总计 120000 次。

方法一：HashMap + TreeMap

思路

先理清题目的意思，朴素的讲就是：一个 key 对应多个值，多个值按照 timestamp 排序，get 操作需返回小于等于 timestamp 的那个值，没有则返回空字符串。

可以使用 HashMap 作为存储结构，定义其 value 是一个能够存储多值的结构即可。可以有如下两种方案：

• HashMap + 二分查找：“提示” 中说明了 timestamp 是严格递增，我们可以直接数组在末尾添加元素即可，然后通过二分查找搜索数组找到对应元素。
• HashMap + TreeMap： 使用 JDK 的 API，TreeMap 提供了向上查找，向下查找的API -> (floorEntry, higherEntry) 等，可以直接使用。

偷个懒，直接用 TreeMap 实现~

参考代码

class TimeMap {

    private final Map<String, TreeMap<Integer, String>> map;

    public TimeMap() {
        map = new HashMap<>();
    }

    public void set(String key, String value, int timestamp) {
        map.computeIfAbsent(key, k -> new TreeMap<>()).put(timestamp, value);
    }

    public String get(String key, int timestamp) {
        Map.Entry<Integer, String> entry = map.getOrDefault(key, new TreeMap<>()).floorEntry(timestamp);
        return entry == null ? "" : entry.getValue();
    }
}

执行结果