Improved task 3433

Author: javadev

src/main/java/g3401_3500/s3433_count_mentions_per_user/Solution.java

@@ -1,49 +1,47 @@
 package g3401_3500.s3433_count_mentions_per_user;
 
-// #Medium #Array #Math #Sorting #Simulation #2025_01_27_Time_29_(100.00%)_Space_45.65_(100.00%)
+// #Medium #Array #Math #Sorting #Simulation #2025_01_28_Time_12_(99.94%)_Space_45.54_(94.71%)
 
+import java.util.ArrayList;
 import java.util.List;
 
 public class Solution {
     public int[] countMentions(int numberOfUsers, List<List<String>> events) {
-        events.sort(
-                (a, b) -> {
-                    int time1 = Integer.parseInt(a.get(1));
-                    int time2 = Integer.parseInt(b.get(1));
-                    if (time1 == time2
-                            && a.get(0).equals("OFFLINE")
-                            && b.get(0).equals("MESSAGE")) {
-                        return -1;
-                    }
-                    return Integer.parseInt(a.get(1)) - Integer.parseInt(b.get(1));
-                });
         int[] ans = new int[numberOfUsers];
-        int[] usertimestamp = new int[numberOfUsers];
-        for (List<String> event : events) {
-            String msg = event.get(0);
-            int time = Integer.parseInt(event.get(1));
-            if (msg.equals("OFFLINE")) {
-                usertimestamp[Integer.parseInt(event.get(2))] = time + 60;
-            } else {
-                String mentionsString = event.get(2);
-                if (mentionsString.equals("ALL")) {
-                    for (int i = 0; i < numberOfUsers; i++) {
-                        ans[i]++;
-                    }
-                } else if (mentionsString.equals("HERE")) {
-                    for (int i = 0; i < numberOfUsers; i++) {
-                        if (usertimestamp[i] <= time) {
-                            ans[i]++;
-                        }
+        List<Integer> l = new ArrayList<>();
+        int c = 0;
+        for (int i = 0; i < events.size(); i++) {
+            String s = events.get(i).get(0);
+            String ss = events.get(i).get(2);
+            if (s.equals("MESSAGE")) {
+                if (ss.equals("ALL") || ss.equals("HERE")) {
+                    c++;
+                    if (ss.equals("HERE")) {
+                        l.add(Integer.parseInt(events.get(i).get(1)));
                     }
                 } else {
-                    for (String id : event.get(2).split(" ")) {
-                        int curr = Integer.parseInt(id.substring(2));
-                        ans[curr]++;
+                    String[] sss = ss.split(" ");
+                    for (int j = 0; j < sss.length; j++) {
+                        int jj = Integer.parseInt(sss[j].substring(2, sss[j].length()));
+                        ans[jj]++;
                     }
                 }
             }
         }
+        for (int i = 0; i < events.size(); i++) {
+            if (events.get(i).get(0).equals("OFFLINE")) {
+                int id = Integer.parseInt(events.get(i).get(2));
+                int a = Integer.parseInt(events.get(i).get(1)) + 60;
+                for (int j = 0; j < l.size(); j++) {
+                    if (l.get(j) >= a - 60 && l.get(j) < a) {
+                        ans[id]--;
+                    }
+                }
+            }
+        }
+        for (int i = 0; i < numberOfUsers; i++) {
+            ans[i] += c;
+        }
         return ans;
     }
 }