2020-09-10

JiayangWu · JiayangWu · commit 33c91a9eb887 · 2020-09-10T23:50:31.000-04:00
diff --git a/1539.第k个缺失的正整数/1539-第k个缺失的正整数.py b/1539.第k个缺失的正整数/1539-第k个缺失的正整数.py
@@ -0,0 +1,14 @@
+class Solution(object):
+    def findKthPositive(self, arr, k):
+        """
+        :type arr: List[int]
+        :type k: int
+        :rtype: int
+        """
+        cnt = 0
+        s = set(arr)
+        for i in range(1, max(arr) + k + 1):
+            if i not in s:
+                cnt += 1
+            if cnt == k:
+                return i
