力扣做题记录——剑2 04.二维数组中的查找

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

[
  [1,   4,  7, 11, 15],
  [2,   5,  8, 12, 19],
  [3,   6,  9, 16, 22],
  [10, 13, 14, 17, 24],
  [18, 21, 23, 26, 30]
]

给定target=5，返回true。
给定target= 20，返回false。
限制

• 0 <= n <= 1000
• 0 <= m <= 1000

思路
  看到这道题，首先最简单的思路自然就是遍历数组，逐个比较。但是，遍历数组这种方法对于任意数组都适用，并且其时间复杂度为O(n*m)。这样，题目中所给出的递增条件就没有用到。那我们回想之前做过的有序数组的查找，就可以想起对于有序数组的查找，可以用二分法完成。在本题中找到位于中心的数当然可行，但是效率太低。因此我们可以选用每一行的最后一个元素来当作“分界线”，从右上向左下遍历，我们将这个元素称为元素a。若元素a大于target，说明target在a的左侧；若元素小于target，说明target在a的右下方。
代码

// 1.遍历
bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) {
    int m = matrix.size();
    if (!m) return false;
    int n = matrix[0].size();
    for (int i = 0; i < m; ++i)
    {
        for (int j = 0; j < n; ++j)
        {
            if (matrix[i][j] == target)
                return true;
        }
    }
    return false;
}

// 2.类二分法
bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) {
	int m = matrix.size();
    if (!m) return false;
    int n = matrix[0].size();
    int i = 0, j = n - 1;
    while (i < m && j >= 0)
    {
        if (matrix[i][j] > target)
            --j;
        else if (matrix[i][j] < target)
            ++i;
        else
            return true;
    }
    return false;
}

  本文题目及部分解答来自力扣：04. 二维数组中的查找