题目来源力扣
题目描述:
创建一个基于时间的键值存储类 TimeMap,它支持下面两个操作:
- set(string key, string value, int timestamp)
存储键 key、值 value,以及给定的时间戳 timestamp。- get(string key, int timestamp)
返回先前调用 set(key, value, timestamp_prev) 所存储的值,其中 timestamp_prev <= timestamp。
如果有多个这样的值,则返回对应最大的 timestamp_prev 的那个值。
如果没有值,则返回空字符串("")。
所有 TimeMap.set 操作中的时间戳 timestamps 都是严格递增的。
审题
考虑到同一个key可能对应多个不同timestamp的value, 因此,我们可以创建键值对<String, TimeStampValue>
,其中TimeStampValue
为自定义类型, 保存了对应的value
与timestamp
. 由于题目保证了set
操作中的timestamps都是严格递增的,因此我们可以使用键值对查找到的当前key
对应的timestamps
对象,然后使用二分查找距离给定timestamp
最近的纪录.
方法一: java算法实现
class TimeMap {
//题目中保证了timestamp插入都是增加的
class TimeStampValue{
private List<String> vals;
private List<Integer> timestamps;
public TimeStampValue(){
vals = new ArrayList<String>();
timestamps = new ArrayList<Integer>();
}
public void set(String value, int timestamp){
vals.add(value);
timestamps.add(timestamp);
}
//使用二分搜索查找
public String get(int timestamp){
int lo = 0;
int hi = timestamps.size();
while(lo < hi){
int mid = lo + (hi-lo) / 2;
if(timestamps.get(mid) == timestamp)
return vals.get(mid);
else if(timestamps.get(mid) > timestamp)
hi = mid;
else
lo = mid+1;
}
if(lo-1 >=0 && lo-1 <= vals.size())
return vals.get(lo-1);
else
return "";
}
}
Map<String, TimeStampValue> map;
public TimeMap() {
map = new HashMap<>();
}
public void set(String key, String value, int timestamp) {
if(map.containsKey(key)){
map.get(key).set(value, timestamp);
}
else{
TimeStampValue tsv = new TimeStampValue();
tsv.set(value, timestamp);
map.put(key, tsv);
}
}
public String get(String key, int timestamp) {
return map.get(key).get(timestamp);
}
}
注:在力扣官方解答中,同时提到java中的TreeMap
支持floorKey
方法,该方法可返回小于等于key
的最大键值, 因此,我们考虑直接使用TreeMap
类代替我们在方法一中自定义的TimeStampValue
类.
方法二:
class TimeMap {
Map<String, TreeMap<Integer, String>> map;
/** Initialize your data structure here. */
public TimeMap() {
map = new HashMap<>();
}
public void set(String key, String value, int timestamp) {
if(map.containsKey(key)){
map.get(key).put(timestamp, value);
}
else{
TreeMap<Integer, String> tm = new TreeMap<>();
tm.put(timestamp, value);
map.put(key, tm);
}
}
public String get(String key, int timestamp) {
Integer tKey = map.get(key).floorKey(timestamp);
return tKey == null ? "" : map.get(key).get(tKey);
}
}
分析方法二,可以发现,尽管我们使用标准库中的TreeMap
类简化了实现,但TreeMap
类并未有效利用题目中给定的条件**所有 TimeMap.set 操作中的时间戳 timestamps 都是严格递增的。**我们直接使用TreeMap
中的标准方法,所以向TreeMap中插入和获取的时间复杂度都是O(lgN), 而在方法一中,插入复杂度为O(1), 获取timestamp对应value的时间复杂度为O(lgN);