剑指Offer -1- 二维数组中的查找 - C++

二维数组中的查找

题目来源： 牛客网 —— 二维数组中的查找

题目描述

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

example:
1 5 10
2 6 11
3 7 12

解题思路：

将用户输入的整数依次与每行的最后一个数（从第一行开始，依次递增） 作比较，如果小于该数，说明在该数的左边，列- - ；如果大于该数，说明在下一行，行++

代码

class Solution {
public:
    bool Find(int target, vector<vector<int> > array) {
        int row = 0;  //行
        int col = array[0].size() - 1;  //列
        int array_row = array.size() - 1;  //二维数组的行数
        while(row <= array_row && col >= 0){  //注意此处的col检测不能遗漏
            if(array[row][col] == target)
                return true;
            // 如果target小于该行的最后一个数，说明在这个数的左边，列--
            else if(array[row][col] > target)
                col --;
            // 如果target大于该行的最后一个数，说明在这个数的下一行，行++
            else
                row ++;
        }
        return false;
    }
};