Create distinct-numbers-in-each-subarray.py

Author: kamyu104

Python/distinct-numbers-in-each-subarray.py

@@ -0,0 +1,25 @@
+# Time:  O(n)
+# Space: O(k)
+
+import collections
+
+
+class Solution(object):
+    def distinctNumbers(self, nums, k):
+        """
+        :type nums: List[int]
+        :type k: int
+        :rtype: List[int]
+        """
+        result = []
+        count = collections.Counter()
+        for i, num in enumerate(nums):
+            count[num] += 1
+            if i >= k:
+                count[nums[i-k]] -= 1
+                if not count[nums[i-k]]:
+                    del count[nums[i-k]]
+            if i+1 >= k:
+                result.append(len(count))
+        return result
+