C++编写二维数组的查找

这是一个很经典的leetcode面试题，要求时间复杂度O(n),空间复杂度O(1)。

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

 

解法1：

我们最简单的求解就是双重for循环然后遍历 数组中的每一个元素是否= target，若等于则返回true。这样时间复杂度是O(m*n) ，m是行的个数，n是列的个数，不符合题目中O(n)的要求。

正确的方法如下

解法2：

主要方法实现：

bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) {
    if (matrix.size() == 0) return false;
    int row = 0;
    int col = matrix[0].size() - 1;

    while (row < matrix.size() && col >= 0) {
        if (matrix[row][col] < target)
        {
            row++;
        }
        else if (matrix[row][col] > target)
        {
            col--;
        }
        else
        {
            return true;
        }
    }
    return false;
}

解题思路：

从左上角 （0，col-1）开始 依次开始比较 ，他题目中的非递减是解题的关键。while循环条件为从右上判断到左下row < matrix.size() && col >= 0。