二进制搜索：无限循环

Question

我正在尝试解决二分搜索问题，但是，我不断遇到无限循环，因为 lo hi 和 mid 的值不会像他们想象的那样改变。我正在尝试搜索函数count_lines(mid,k) >= n 的最小中间值。我花了几个小时调试这段代码，但我仍然无法弄清楚lohi 和mid 的值有什么问题。有人可以看看我的代码并告诉我为什么会这样吗？非常感谢！

失败的测试用例：n=9k=2 这是我的代码：

   def count_lines(v,k):
   ...
   ...
   return count_lines #count_lines is an integer` 


   def binarySearch_v(n, k):
   lo = 0
   hi = n
   mid = (lo + hi)//2
   while (lo <= hi):
     if (count_lines(mid, k) < n):
        lo = mid
     elif (count_lines(mid, k) > n):
        hi = mid
     elif (count_lines(mid, k) >= n and count_lines(mid-1, k) < n):
        return mid
     mid = (lo + hi) // 2

    def main():
      print(binarySearch_v(n,k)) # Failed at n=9 & k=2
    main()

Answer 1

这是您的终止条件：

while (lo <= hi):

这里是lo 和hi 可能改变值的地方：

lo = mid
...
hi = mid

合乎逻辑的结论是，如果这段代码循环的次数足够多，您最终会得到lo == hi。然后mid 将设置为两者的平均值，等于它们两者，这就是所有三个变量都将保留的地方。因此，循环不会终止。有两种可能的解决方案：要么更改重新分配lo 或hi 的方式，要么将终止条件更改为< 而不是<=。