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

题目描述：

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

解题思路：

1）输入目标值与待查找数组；
2）获取待查找数组的行数rows与列数cols，若rows值为0，则返回flase，函数结束；
3）将目标值与待查找数组的第一行最后一列数a比较；
4）若目标值与a相等，则返回目标值；若目标值大于a，则与下一行的最后一列数比较；若目标值小于a，则于该行的前一列数b比较；
5）若目标值不在数组中，返回false，函数结束；

代码：

class Solution {
public:
    bool Find(int target, vector<vector<int> > array) {
        // 获取array的长度使用array.size();
        // 算法思路：
        // 输入目标值与数组，第一步就应该判断数组是否是二维数组，
        // 然后将目标值与数组的第一行最后一列的数a进行比较
        // 若目标值大于a，则向下移动一行进行搜索，与下一行的最后一列数b比较
        // 若目标值小于a，则向前移动一列进行搜索，与前一列的数c比较
        // for(int row = 0,col = cols-1;row <= rows-1 && col >= 0;)
        //     if( target == array[row][col])
        //        return target;
        //     else if(target < array[row][col])
        //         col--;
        //     else if(target > array[row][col])
        //         row++;
        // return false;
        int rows = array.size();
        int cols = array[0].size();
        for(int row = 0, col = cols-1;row <= rows-1 && col >= 0;)
        {
            if( target == array[row][col])
                 return target;
            else if(target < array[row][col])
                col--;
            else if(target > array[row][col])
                 row++;
        }
        return false;
    }
};