Google
359 Logger Rate Limiter

1. Question

Design a logger system that receive stream of messages along with its timestamps, each message should be printed if and only if it is not printed in the last 10 seconds.
Given a message and a timestamp (in seconds granularity), return true if the message should be printed in the given timestamp, otherwise returns false.
It is possible that several messages arrive roughly at the same time.
Example:
1
Logger logger = new Logger();
2
3
// logging string "foo" at timestamp 1
4
logger.shouldPrintMessage(1, "foo"); returns true;
5
6
// logging string "bar" at timestamp 2
7
logger.shouldPrintMessage(2,"bar"); returns true;
8
9
// logging string "foo" at timestamp 3
10
logger.shouldPrintMessage(3,"foo"); returns false;
11
12
// logging string "bar" at timestamp 8
13
logger.shouldPrintMessage(8,"bar"); returns false;
14
15
// logging string "foo" at timestamp 10
16
logger.shouldPrintMessage(10,"foo"); returns false;
17
18
// logging string "foo" at timestamp 11
19
logger.shouldPrintMessage(11,"foo"); returns true;
Copied!

2. Implementation

(1) HashMap
1
class Logger {
2
Map<String, Integer> map;
3
4
/** Initialize your data structure here. */
5
public Logger() {
6
map = new HashMap<>();
7
}
8
9
/** Returns true if the message should be printed in the given timestamp, otherwise returns false.
10
If this method returns false, the message will not be printed.
11
The timestamp is in seconds granularity. */
12
public boolean shouldPrintMessage(int timestamp, String message) {
13
if (!map.containsKey(message) || (timestamp - map.get(message)) >= 10) {
14
map.put(message, timestamp);
15
return true;
16
}
17
return false;
18
}
19
}
20
21
/**
22
* Your Logger object will be instantiated and called as such:
23
* Logger obj = new Logger();
24
* boolean param_1 = obj.shouldPrintMessage(timestamp,message);
25
*/
Copied!
(2) Set + Deque
思路: 上一种方法由于不停将message存在hashmap中,所以空间消耗很大。为了优化空间,我们采用Deque,当在deque的头比timestamp小于10秒之内时,我们将其从deque remove
1
class Logger {
2
class Log {
3
int time;
4
String message;
5
6
public Log(int time, String message) {
7
this.time = time;
8
this.message = message;
9
}
10
}
11
12
private Set<String> set;
13
private Deque<Log> deque;
14
15
/** Initialize your data structure here. */
16
public Logger() {
17
set = new HashSet();
18
deque = new ArrayDeque();
19
}
20
21
/** Returns true if the message should be printed in the given timestamp, otherwise returns false.
22
If this method returns false, the message will not be printed.
23
The timestamp is in seconds granularity. */
24
public boolean shouldPrintMessage(int timestamp, String message) {
25
while (!deque.isEmpty() && (deque.peek().time <= timestamp - 10)) {
26
Log oldLog = deque.remove();
27
set.remove(oldLog.message);
28
}
29
30
if (!set.contains(message)) {
31
set.add(message);
32
deque.add(new Log(timestamp, message));
33
return true;
34
}
35
return false;
36
}
37
}
38
39
/**
40
* Your Logger object will be instantiated and called as such:
41
* Logger obj = new Logger();
42
* boolean param_1 = obj.shouldPrintMessage(timestamp,message);
43
*/
Copied!

3. Time & Space Complexity

HashMap: 时间复杂度O(1), 空间复杂度O(n)
Set + Deque: 时间复杂度O(1), 空间复杂度O(n)