【牛客刷题系列】二分查找常见面试题总结

二分查找常规实现

二分查找也称为折半查找，每次都能将查找区间减半，这种折半特性的算法时间复杂度为 O(logN)。

Input : [1,2,3,4,5]
key : 3
return the index : 2

int binarySearch(int* nums, int length, int key) {
    int l = 0, h = length - 1;
    while (l <= h) {
        int m = l + (h - l) / 2;
        if (nums[m] == key) {
            return m;
        } else if (nums[m] > key) {
            h = m - 1;
        } else {
            l = m + 1;
        }
    }
    return -1;
}

二维数组中的查找

在一个二维数组中（每个一维数组的长度相同），每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。

示例1

输入 7,[[1,2,8,9],[2,4,9,12],[4,7,10,13],[6,8,11,15]]
返回值 true

实现

方法1.暴力搜索：时间复杂度O（n^2）,直接全部遍历，因为最坏情况下，数组中的元素都需要遍历一次

class Solution {
public:
    bool Find(int target, vector<vector<int> > array) {
        int i=array.size();
        int j=array[0].size();
        for(int n=0;n<i;n++)
        {
            for(int m=0;m<j;m++)
            {
                if(array[n][m]==target)
                    return true;
            }
        }
        return false;
    }
};

方法2:二分查找

分析：对于方法一，此题有额外信息没有利用上，数组从左到右递增，从上到下递增。有序的数组很显然应该想到二分。那么应该如何二分呢？

假设arr数组，val，tar如下图所示：
如果我们把二分值定在右上角或者左下角，就可以进行二分。这里以右上角为例，左下角可自行分析：

1）我么设初始值为右上角元素，arr[0][5] = val，目标tar = arr[3][1]
2）接下来进行二分操作：
3）如果val == target,直接返回
4）如果 tar > val, 说明target在更大的位置，val左边的元素显然都是 < val，间接 < tar，说明第 0 行都是无效的，所以val下移到arr[1][5]
5）如果 tar < val, 说明target在更小的位置，val下边的元素显然都是 > val，间接 > tar，说明第 5 列都是无效的，所以val左移到arr[0][4]
6）继续步骤2）

复杂度分析
时间复杂度：O(m+n) ，其中m为行数，n为列数，最坏情况下，需要遍历m+n次。
空间复杂度：O(1)

class Solution {
public:
    bool Find(int target, vector<vector<int> > array) {
     int i = array.size();
     int j = array[0].size();
     int n = 0, m = j-1; // 右上角元素
     while (n<i && m>=0) {
         if (target == array[n][m]) {
             return true;
         }
         else if (target > array[n][m])
             ++n;
         else 
             --m;
     }
     return false;
    }
};