题目描述

在一个二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。

请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。

样例

输入数组：

[
  [1,2,8,9]，
  [2,4,9,12]，
  [4,7,10,13]，
  [6,8,11,15]
]

如果输入查找数值为7，则返回true，

如果输入查找数值为5，则返回false。

思路

本题做法是在数组的右上角起手，不断缩小数组搜索范围

• 若目标小于右上角，则可以排除本列；(因为本列中其余的数都比右上角要大)
• 若目标大于右上角，则可以排除本行。(因为本行中其余的数都比右上角要小)

代码

class Solution {
public:
    bool searchArray(vector<vector<int>> array, int target) {
        auto rows = array.size();
        if(rows == 0) return false;
        auto cols = array[0].size();

        unsigned int row = 0, col = cols - 1;
        while(row < rows && col < cols) {
            if(array[row][col] < target) row++;
            else if(array[row][col] > target) col--;
            else return true;
        }
        return false;
    }
};