【剑指offer】二维数组中的查找

题目描述

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

解题思路

这个二维数组有很好的结构，简单画个图。

思路一: 二分查找

二分查找(折半查找)要求待查找的表/数组满足两个条件：
(1) 查找表/数组必须使用顺序的存储结构；
(2) 查找表/数组必须按照关键字大小有序排列。

因此，把这道题中二维数组的每一行看成有序递增的数组，对每一行进行二分查找操作，就可以得到答案，时间复杂度是 O(nlogn)。

其中，还有一个小小trick，当判断出 target 小于二分查找中间值 array[i][mid] 时，利用二维表的结构，可以知道，在当前 array[i][mid] 值及其右侧所有值都大于 target 值，也就是说，列下标大于等于当前mid的值都不符合条件，无需再搜索，可直接将 cols 减少到 mid 列，即 cols=mid，这样可以减少二分查找的空间。

实现代码如下：

class Solution {
public:
    bool Find(int target, vector<vector<int> > array) {
        int rows = array.size();
        int cols = array[0].size();
        for(int i=0;i<rows;i++){
            int low=0,high=cols-1;
            int mid;
            while(low<=high){
                mid = (low+high)/2;
                if(target<array[i][mid]){
                    high = mid-1;
                    cols = mid;
                }
                else if(target>array[i][mid]){
                    low = mid+1;
                }
                else{
                    return true;
                }
            }
        }
        return false;
    }
};

思路二：

利用二维数组由上到下，由左到右递增的规律，从左下角的元素开始搜索，当target小于当前元素时，上移col–；当target大于当前元素时，右移row++。

实现代码：

class Solution {
public:
    bool Find(int target, vector<vector<int> > array) {
        int rows = array.size();
        int cols = array[0].size();
        int i = rows-1, j = 0;
        while(i>=0 && j<cols){
            if(target<array[i][j]){
                i--;
            }
            else if(target>array[i][j]){
                j++;
            }
            else{
                return true;
            }
        }
        return false;
    }
};

参考：

1. https://www.nowcoder.com/questionTerminal/abc3fe2ce8e146608e868a70efebf62e