optimize binary_search

polixir · Dec 31, 2020 · e2bcfd8 · e2bcfd8
1 parent 1b7d88a
commit e2bcfd8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/zoopt/algos/opt_algorithms/racos/sracos.py b/zoopt/algos/opt_algorithms/racos/sracos.py
@@ -143,7 +143,7 @@ def binary_search(self, iset, x, begin, end):
             return end + 1
         if end == begin + 1:
             return end
-        mid = (begin + end) // 2
+        mid = begin + (end - begin) // 2
         if x_value <= iset[mid].get_value():
             return self.binary_search(iset, x, begin, mid)
         else:
@@ -369,7 +369,7 @@ def binary_search(self, iset, x, begin, end):
             return end + 1
         if end == begin + 1:
             return end
-        mid = (begin + end) // 2
+        mid = begin + (end - begin) // 2
         if x_value <= iset[mid].get_value():
             return self.binary_search(iset, x, begin, mid)
         else: