Feb 7

Author: siddydutta

2025-02-February-LeetCoding-Challenge/Find the Number of Distinct Colors Among the Balls.py

@@ -0,0 +1,28 @@
+class Solution:
+    def queryResults(self, _: int, queries: list[list[int]]) -> list[int]:
+        ball_color = dict()
+        color_count = dict()
+        result = []
+        for ball, color in queries:
+            if ball in ball_color:
+                color_count[ball_color[ball]] -= 1
+                if color_count[ball_color[ball]] == 0:
+                    del color_count[ball_color[ball]]
+            ball_color[ball] = color
+            color_count[color] = color_count.get(color, 0) + 1
+            result.append(len(color_count))
+        return result
+
+
+def main():
+    limit = 4
+    queries = [[1, 4], [2, 5], [1, 3], [3, 4]]
+    assert Solution().queryResults(limit, queries) == [1, 2, 2, 3]
+
+    limit = 4
+    queries = [[0, 1], [1, 2], [2, 2], [3, 4], [4, 5]]
+    assert Solution().queryResults(limit, queries) == [1, 2, 2, 3, 4]
+
+
+if __name__ == '__main__':
+    main()

2025-02-February-LeetCoding-Challenge/README.md

@@ -10,7 +10,7 @@
 | February 04 | [1800. Maximum Ascending Subarray Sum](https://leetcode.com/problems/maximum-ascending-subarray-sum/) | Easy | Solved |
 | February 05 | [1790. Check if One String Swap Can Make Strings Equal](https://leetcode.com/problems/check-if-one-string-swap-can-make-strings-equal/) | Easy | Solved |
 | February 06 | [1726. Tuple with Same Product](https://leetcode.com/problems/tuple-with-same-product/) | Medium | Solved |
-| February 07 | []() |  |  |
+| February 07 | [3160. Find the Number of Distinct Colors Among the Balls](https://leetcode.com/problems/find-the-number-of-distinct-colors-among-the-balls/) | Medium | Solved |
 | February 08 | []() |  |  |
 | February 09 | []() |  |  |
 | February 10 | []() |  |  |
@@ -38,5 +38,5 @@
 | Level  | Problems | Solved | Unsolved |
 | ---    | --- | --- | --- |
 | Easy   | 5 | 5 | 0 |
-| Medium | 1 | 1 | 0 |
+| Medium | 2 | 2 | 0 |
 | Hard   | 0 | 0 | 0 |