Time: 153 ms (6.25%), Space: 64.6 MB (5.56%) - LeetHub

Amit-S-Sahu · Oct 28, 2024 · e552a99 · e552a99
1 parent 09e956f
commit e552a99
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/2501-longest-square-streak-in-an-array/2501-longest-square-streak-in-an-array.java b/2501-longest-square-streak-in-an-array/2501-longest-square-streak-in-an-array.java
@@ -0,0 +1,21 @@
+class Solution {
+    public int longestSquareStreak(int[] nums) {
+        TreeSet<Integer> set = new TreeSet<>();
+        for(int num : nums) {
+            set.add(num);
+        }
+
+        int ans = 1;
+        while(!set.isEmpty()) {
+            int curr = set.first();
+            int streak = 0;
+            while(!set.isEmpty() && set.contains(curr)) {
+                set.remove(curr);
+                curr = curr * curr;
+                streak++;
+            }     
+            ans = Math.max(streak, ans);
+        }
+        return ans == 1 ? -1 : ans;
+    }
+}