362. Design Hit Counter

Design a hit counter which counts the number of hits received in the past 5 minutes.

Each function accepts a timestamp parameter (in seconds granularity) and you may assume that calls are being made to the system in chronological order (ie, the timestamp is monotonically increasing). You may assume that the earliest timestamp starts at 1.

It is possible that several hits arrive roughly at the same time.

Example:

HitCounter counter = new HitCounter();

// hit at timestamp 1.
counter.hit(1);

// hit at timestamp 2.
counter.hit(2);

// hit at timestamp 3.
counter.hit(3);

// get hits at timestamp 4, should return 3.
counter.getHits(4);

// hit at timestamp 300.
counter.hit(300);

// get hits at timestamp 300, should return 4.
counter.getHits(300);

// get hits at timestamp 301, should return 3.
counter.getHits(301); 

Follow up:
What if the number of hits per second could be very large? Does your design scale?

Credits:
Special thanks to @elmirap for adding this problem and creating all test cases.

开始用queue存hit的序列，用hashmap存timestamp的个数，gethits时注意消除过期的timestamp。代码如下：

public class HitCounter {
    //HashSet<Integer> hs;
    Queue<Integer> queue;
    HashMap<Integer, Integer> map;
    int total;

    /** Initialize your data structure here. */
    public HitCounter() {
        //hs = new HashSet<Integer>();
        queue = new LinkedList<Integer>();
        map = new HashMap<Integer, Integer>();
    }
    
    /** Record a hit.
        @param timestamp - The current timestamp (in seconds granularity). */
    public void hit(int timestamp) {
        total ++;
        //hs.add(timestamp);
        queue.offer(timestamp);
        map.put(timestamp, map.getOrDefault(timestamp, 0) + 1);
    }
    
    /** Return the number of hits in the past 5 minutes.
        @param timestamp - The current timestamp (in seconds granularity). */
    public int getHits(int timestamp) {
        while (!queue.isEmpty()) {
            int time = queue.peek();
            if (time + 300 <= timestamp) {
                total -= map.getOrDefault(time, 0);
                map.remove(time);
                queue.poll();
            } else {
                break;
            }
        }
        return total;
    }
}

/**
 * Your HitCounter object will be instantiated and called as such:
 * HitCounter obj = new HitCounter();
 * obj.hit(timestamp);
 * int param_2 = obj.getHits(timestamp);
 */

上面的方法显然是想多了，只用queue就能解题。代码如下：

public class HitCounter {
    Queue<Integer> hit;
    /** Initialize your data structure here. */
    public HitCounter() {
        hit = new LinkedList();
    } 
    
    /** Record a hit.
        @param timestamp - The current timestamp (in seconds granularity). */
    public void hit(int timestamp) {
        hit.offer(timestamp);
    }
    
    /** Return the number of hits in the past 5 minutes.
        @param timestamp - The current timestamp (in seconds granularity). */
    public int getHits(int timestamp) {
        while(hit.size() > 0&&hit.peek() <= timestamp - 300){
            hit.remove();
        }
        return hit.size();
    }
}