剑指Offer(第2版)——面试题4：二维数组中的查找

题目：

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

1.解题思路：

• 首先选取数组中右上角的数字。如果该数字等于要查找的数字，查找过程结束；
• 如果该数字大于要查找的数组，剔除这个数字所在的列；
• 如果该数字小于要查找的数字，剔除这个数字所在的行。
• 也就是说如果要查找的数字不在数组的右上角，则每一次都在数组的查找范围中剔除一行或者一列，这样每一步都可以缩小查找的范围，直到找到要查找的数字，或者查找范围为空。

注：同理也可以取左下角的数字进行判断。

2.代码：

C++

class Solution {
public:
    //形参vector<vector<int> >array表达的是array是一个二维的int型的vector（实际上相当于一个二维数组）
    bool Find(int target, vector<vector<int> > array) {
        int rows = array.size(); //二维数组的行数
        int cols = array[0].size(); //二维数组第一行的列数
        if(!array.empty() && rows > 0 && cols > 0){
            int row = 0;
            int col = cols - 1; //选取右上角数字
            while(row < rows && col >= 0){
                if(array[row][col] == target){
                    return true;
                }
                else if(array[row][col] > target){
                    --col;
                }
                else{
                    ++row;
                }
            }
        }
        return false;
    }
};