【Leetcode Medium】1348. Tweet Counts Per Frequency

原题链接

题目描述

记录推文的name和发布time，并返回要求的[start, end)区间内有多少指定的推文（区间按要求划分）

我的思路

这题最重要的是用什么数据结构来存推文name及发布时间。一开始想了两种数据结构，一个是HashMap<String, Integer>，但是由于hashmap是无序的，计算次数时搜索起来会比较麻烦；第二个是int数组，但只能方便记录次数不能对应记录name。于是得出结论，发布时间的记录一定需要一个有序的数据结构。
参考了discussion中的方法之后采用了HashMap<String, TreeMap<Integer, Integer>>
TreeMap有一个很方便的方法subMap(start, end)，可以返回从start到end的视图。
当然也可以不用TreeMap而使用ArrayList，用二分法找到start和end time的index就可以。
code：

class TweetCounts {
    
    private Map<String, TreeMap<Integer, Integer>> record;

    public TweetCounts() {
        record = new HashMap<>();
    }
    
    public void recordTweet(String tweetName, int time) {
        record.putIfAbsent(tweetName, new TreeMap<Integer, Integer>());
        TreeMap<Integer, Integer> tm = record.get(tweetName);
        tm.put(time, tm.getOrDefault(time, 0) + 1);
    }
    
    public List<Integer> getTweetCountsPerFrequency(String freq, String tweetName, int startTime, int endTime) {
        int split = 60;
        if (freq.equals("hour")) {
            split = 60 * 60;
        }
        else if (freq.equals("day")) {
            split = 24 * 60 * 60;
        }
        
        List<Integer> res = new ArrayList<>();
        TreeMap<Integer, Integer> tm = record.get(tweetName);
        
        if (!record.containsKey(tweetName))
            return res;
        
        for (int start = startTime; start <= endTime; start += split) {
            int count = 0;
            int end = Math.min(endTime + 1, start + split);
            
            Map<Integer, Integer> sub = tm.subMap(start, end);
            
            for (int val : sub.values())
                count += val;
            
            res.add(count);
        }
        
        return res;
    }
}