[剑指Offer] 二维数组中的查找

题目

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

分析

如果选择在中间选择一个数与target进行比较并以矩形的方式拓展搜索范围，如果不相等，那么剩下的搜索范围就会有重叠，所以这种方案比较难以继续往下分析。

于是，根据数组的排列特点，选择右上角或左下角的数与target进行比较。

假设选择右上角的数，如果相等，那么就搜索到了；如果比target大，那么就舍弃当前这个数所在的列；如果比target小，那么就舍弃当前这个数所在的行。 以此来减小搜索范围。

【补充】

vector表示的二维数组，例如：vector<vector<int> > vec，

• 判断vec对象是否为空的方法：vec.empty()；
• 获取行的方式：vec.size();
• 获取列的方式：vec[0].size()。

代码

class Solution {
public:
    bool Find(int target, vector<vector<int> > array) {
        bool found = false;
        if(array.empty() || array.size() == 0) return found;
        int rows = array.size();//获取二维数组的行数
        int columns = array[0].size(); //获取二维数组的列数
        int row = 0;
        int column = columns - 1;
        while(row < rows && column >= 0) {
            if(array[row][column] == target) {
                found = true;
                break;
            } else if(array[row][column] > target) {
                --column;
            } else {
                ++row;
            }
        }
        return found;
    }
};