推文计数

/* 
思路：
1. 创建 TweetCounts 类，其包含记录推文时间的记录。
2. recordTweet 方法用于记录推文的 tweetName 和时间。
3. getTweetCountsPerFrequency 方法根据给定的频率（minute、hour、day）统计指定时间范围内的推文数量。
4. 使用 TreeMap 数据结构存储推文记录，便于按时间排序和查询。 
5. 利用 Java Stream API 处理和统计推文数量。 
*/
import java.util.*;
import java.util.stream.Collectors;

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

    public TweetCounts() {
        tweetRecords = new HashMap<>();
    }

    public void recordTweet(String tweetName, int time) {
        tweetRecords.putIfAbsent(tweetName, new TreeMap<>());
        TreeMap<Integer, Integer> times = tweetRecords.get(tweetName);
        times.put(time, times.getOrDefault(time, 0) + 1);
    }

    public List<Integer> getTweetCountsPerFrequency(String freq, String tweetName, int startTime, int endTime) {
        if (!tweetRecords.containsKey(tweetName)) {
            return Collections.emptyList();
        }
        int interval = getInterval(freq);
        TreeMap<Integer, Integer> times = tweetRecords.get(tweetName);
        return times.subMap(startTime, true, endTime, true).entrySet().stream()
            .collect(Collectors.groupingBy(e -> (e.getKey() - startTime) / interval, Collectors.summingInt(Map.Entry::getValue)))
            .values().stream()
            .collect(Collectors.toList());
    }

    private int getInterval(String freq) {
        switch (freq) {
            case "minute": return 60;
            case "hour": return 3600;
            case "day": return 86400;
            default: throw new IllegalArgumentException("Invalid frequency");
        }
    }
}