leetcode:基于时间的键值存储类

题目来源力扣

题目描述:

创建一个基于时间的键值存储类 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 的那个值。
   如果没有值，则返回空字符串（""）。
   所有 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);