garciparedes
diff --git a/‎problems/2200_find_all_k_distant_indices_in_an_array.py
Lines changed: 41 additions & 0 deletions b/‎problems/2200_find_all_k_distant_indices_in_an_array.py
Lines changed: 41 additions & 0 deletions
@@ -0,0 +1,41 @@
+
+
+class Solution:
+    def findKDistantIndices(self, nums: List[int], key: int, k: int) -> List[int]:
+        key_indexes = list()
+        for i in range(len(nums)):
+            if nums[i] == key:
+                key_indexes.append(i)
+                
+        ans = list()
+        for i in range(len(nums)):
+            if self._is_k_distant(key_indexes, i, k):
+                ans.append(i)
+                
+        return ans
+    
+    @staticmethod
+    def _is_k_distant(indexes: list[int], i: int, k: int) -> bool:
+        
+        left, right = 0, len(indexes) - 1
+        while left < right:
+            mid = ceil((left + right) /  2)
+            
+            if indexes[mid] > i:
+                right = mid - 1
+            else:
+                left = mid
+                
+        if abs(indexes[left] - i) <= k:
+            return True
+        
+        if left < len(indexes) - 1:
+            left += 1
+            return abs(indexes[left] - i) <= k
+        
+        return False
+        
+        
+        
+