二分查找：如何用最省内存的方式实现快速查找功能？

news2024/11/26 10:30:30

文章来源于极客时间前google工程师−王争专栏。

有序数据集合的查找算法：二分查找(Binary Search)算法，也叫折半查找算法。二分查找的思想非常简单，但是难掌握好，灵活运用更加困难。

问题：假设有1000万个整数数据，每个数据占8个字节，如何设计数据结构和算法，快速判断某个整数是否出现在这1000万数据中？这个功能不能占用太多的内存空间，最多不超过100MB，如何解决？

无处不在的二分思想

随机写一个0~99之间的数字，猜数字。（如果猜测范围的数字有偶数个，中间数有两个，就选择较小的那个）

0~999的数字，最多也只要10次就能猜中。

开发场景：假设有1000条订单数据，已经按照订单金额从小到大排序，每个订单金额都不同，并且最小单位是元。我们现在想知道是否存在金额等于19元的订单。如果存在，则返回订单数据，如果不存在则返回null。

最简单的办法是从第一个开始，一个一个遍历。最坏情况下需要1000次。用二分查找能不能更快速地解决呢？

假设只有10个订单，订单金额分别是：8，11，19，23，27，33，45，55，67，98。

二分查找针对的是一个有序的数据集合，查找思想有点类似分治思想。每次都通过跟区间的中间元素对比，将待查找的区间缩小为之前的一半，知道找到要查找的元素，或者区间被缩小为0。

O(logn)惊人的查找速度

二分查找是一种非常高效的查找算法，我们来分析一下它的时间复杂度。

假设数据大小为n，每次查找数据都会缩小为原来的一半，也就是会除以2。最坏情况下，直到查找区间被缩小为空，才停止。

可以看出来，这是一个等比数列。其中n/2k=1时，k的值就是总共缩小的次数。而每一次缩小操作只涉及两个数据的大小比较，所以，经过了 k 次区间缩小操作，时间复杂度就是O(k)。通过n/2k=1，我们可以求得 k=log2n，所以时间复杂度就是 O(logn)。

二分查找是目前为止遇到的第一个时间复杂度为O(logn)的算法。后面的堆、二叉树等操作，它们的时间复杂度也是O(logn)。O(logn)这种对数时间复杂度，是一种及其高效的时间复杂度，有的时候甚至比时间复杂度是常量级O(1)的算法还要高效。

logn是一个非常“恐怖”的数量级，即便n非常非常大，对应的logn也很小。比如n等于2的32次方，这个数很大，大约是42亿。也就是说，如果在42亿个数据中用二分查找一个数据，最多需要比较32次。

O(1)有可能表示的是一个非常大的常量值，比如O(1000)，O(10000)。所以，常量级时间复杂度的算法有时候可能还没有O(logn)的算法执行效率高。

对数对应的就是指数。“阿基米德与国王下棋的故事”

二分查找的递归与非递归实现

简单的二分查找不难写，二分查找的变体问题，才是真正烧脑的。

最简单的情况就是有序数组中不存在重复元素，代码实现如下：

public int bsearch(int[] a, int n, int value) {
  int low = 0;
  int high = n - 1;

  while (low <= high) {
    int mid = (low + high) / 2;
    if (a[mid] == value) {
      return mid;
    } else if (a[mid] < value) {
      low = mid + 1;
    } else {
      high = mid - 1;
    }
  }
  return -1;
}

容易出错的3个地方：

1.循环退出条件

是low<=high，而不是low<high

2.mid的取值

mid=(low+high)/2这种写法是有问题的。如果low和high比较大，两者之和就有可能溢出。

改进的方法是mid=low+(high-low)/2

如果性能做到极致的话，更进一步可以将除以2转化为位运算mid=((high-low)>>1)。因为相比除法运算来说，计算机处理位运算要快得多。

3.low和high的更新

low=mid+1,high=mid-1。如果写成low=mid或者high=mid就可能发生死循环。

比如high=3，low=3时，如果a[3]不等于value，就会导致一直循环不退出。

实际上，二分查找除了用循环来实现，还可以用递归来实现。

// 二分查找的递归实现
public int bsearch(int[] a, int n, int val) {
  return bsearchInternally(a, 0, n - 1, val);
}

private int bsearchInternally(int[] a, int low, int high, int value) {
  if (low > high) return -1;

  int mid =  low + ((high - low) >> 1);
  if (a[mid] == value) {
    return mid;
  } else if (a[mid] < value) {
    return bsearchInternally(a, mid+1, high, value);
  } else {
    return bsearchInternally(a, low, mid-1, value);
  }
}