二分法常见套路

众所周知，二分法能非常高效地从有序数组中找出特定的元素，时间复杂度为 O(logN)。

一般而言，遇到要求 O(logN) 时间复杂度的搜索问题，十有八九会用到二分查找。

注意：求二分法的中点时注意防止溢出，两个 32 位正整数相加可能会溢出。

下面整理下二分法常见的套路：

套路一：搜索数组中等于某个元素的位置

这种问题基本就是套二分法的模版，二分法模版整理如下：

public int binarySearch(int[] nums, int target) {
        int left = 0, right = nums.length - 1;
        int mid;
        while (left <= right) {
            mid = ((right - left) >> 1) + left; // 这里不用 (left + right) / 2 是为了防止溢出
            if (nums[mid] == target) {
                return mid;
            } else if (nums[mid] < target) {
                left = mid + 1;
            } else {
                right = mid - 1;
            }
        }
        return -1; // 没找到
}

这类型的题举例如下：

LeetCode 74 题：搜索二维矩阵

题目： 编写一个高效的算法来判断 m x n 矩阵中，是否存在一个目标值。该矩阵具有如下特性： 每行中的整数从左到右按升序排列。 每行的第一个整数大于前一行的最后一个整数。

解析： 由题目可以看出，每一行的第一个数大于上一行最后一个数，如果将二维数组看成一个大的一维数组，那么整个数组是严格升序的，可以直接套用模板，唯一的问题是要将一维数组的索引，转化为二维数组的索引，这个转化是很简单的：假设一维数组的索引是 i，二维数组的大小是 m x n，则 i 在二维数组中的位置是：i / n, i % n。

答案：

    public boolean searchMatrix(int[][] matrix, int target) {
        int m = matrix.length, n = matrix[0].length;
        int left = 0, right = m * n - 1;
        int mid;
        while (left <= right) {
            mid = ((right - left) >> 1) + left;
            if (matrix[mid/n][mid%n] == target) {
                return true;
            }
            if (matrix[mid/n][mid%n] < target) {
                left = mid + 1;
            } else {
                right = mid - 1;
            }
        }
        return false;