refactor 359

fishercoder1534 · fishercoder1534 · commit 2624c2d6fa98 · 2021-07-12T11:16:34.000-07:00
diff --git a/src/main/java/com/fishercoder/solutions/_359.java b/src/main/java/com/fishercoder/solutions/_359.java
@@ -11,24 +11,14 @@ public static class Solution1 {
         class Logger {
 
             private Map<String, Integer> map;
-            private Set<String> set;
 
-            /**
-             * Initialize your data structure here.
-             */
             public Logger() {
-                map = new HashMap<String, Integer>();
-                set = new HashSet<String>();
+                map = new HashMap<>();
             }
 
-            /**
-             * Returns true if the message should be printed in the given timestamp, otherwise returns
-             * false. The timestamp is in seconds granularity.
-             */
             public boolean shouldPrintMessage(int timestamp, String message) {
-                if (!set.contains(message)) {
+                if (!map.containsKey(message)) {
                     map.put(message, timestamp);
-                    set.add(message);
                     return true;
                 } else {
                     if (timestamp - map.get(message) < 10) {
