剑指offer 面试题4 二维数组中的查找

1.题意

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

 

2.思路

从矩阵的右上角开始搜索，假设右上角的数字为m，目标值为target，有如下三种情况：

(1) m == target，查找成功，返回true;

(2) m > target，每一列都按照从上到下递增的顺序排序，m所在列的数字肯定都大于target，因此将m所在列的数字剔除。

(3) m < target，每一行都按照从左到右递增的顺序排序，m所在行的数字肯定都小于target，因此将m所在行的数字剔除。

这样每一步都可以缩小查找范围，从余下的数字中继续查找。

 

时间复杂度：

3.实现

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