【剑指 offer 66题】二维数组查找

题目描述

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

思路

思路1

把每一行看成有序递增的数组，利用二分查找，通过遍历每一行得到答案，时间复杂度是 $\mathcal{O}(mlogn)$。

思路2

利用二维数组由上到下，由左到右递增的规律，那么选取右上角或者左下角的元素 a[row][col] 与target进行比较，当target小于元素a[row][col]时，那么target必定在元素a所在行的左边,即col–；当target大于元素a[row][col]时，那么target必定在元素a所在列的下边,即row++；时间复杂度是 $\mathcal{O}(m+n)$。

代码

/*
思路2
*/
class Solution {
public:
    bool Find(int target, vector<vector<int> > array) {
        int rows = array.size(); // 行数
        int cols = array[0].size(); //列数
        //从左上角开始
        int i=0; 
        int j=cols-1;
        //
        while(i<rows && j>=0){ //边界
                if(target == array[i][j])
                    return true;
                else if (target > array[i][j]) 
                    ++i;
                else
                    --j;
        }
        return false; //超出边界则返回失败
    }
};

扩展

vector

链接：C++ vector 容器浅析.