change first occurrence algorithm

dori-dev · dori-dev · commit 3a65a153738f · 2022-08-03T20:54:34.000+04:30
diff --git a/search/first_occurrence.py b/search/first_occurrence.py
@@ -9,7 +9,7 @@
 
 def first_occurrence(array: list, target: Any) -> int:
     low, high = 0, len(array) - 1
-    while low < high:
+    while low <= high:
         middle = (low + high) // 2
         if array[middle] < target:
             low = middle + 1
@@ -23,3 +23,5 @@ def first_occurrence(array: list, target: Any) -> int:
 if __name__ == "__main__":
     print(first_occurrence([2, 2, 2, 3, 3, 4, 4, 5, 5, 5], 3) == 3)
     print(first_occurrence([2, 2, 2, 3, 3, 4, 4, 5, 5, 5], 2) == 0)
+    print(first_occurrence([2, 2, 2, 3, 3, 4, 4, 5, 5, 5], 4) == 5)
+    print(first_occurrence([2, 2, 2, 3, 3, 4, 4, 5, 5, 5], 5) == 7)
