原题链接
题目描述
记录推文的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;
}
}