剑指 Offer 04. 二维数组中的查找 | 算法

92 阅读2分钟

剑指 Offer 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

方法一:使用js特有的数组方法实现

array.flat() 扁平化嵌套实现

flat() 方法会按照一个可指定的深度递归遍历数组,并将所有元素与遍历到的子数组中的元素合并为一个新数组返回。

参数:默认值为1,infinity为任意深度的嵌套数组。

  var findNumberIn2DArray = function(matrix, target) {
    return matrix.flat().includes(target);
  };

方法二:选取数组中合适的位置与目标值比较,逐步缩减范围。

这个合适的位置可以选择数组右上角/左下角位置(本题选用右上角位置)。这一种方式最坏情况下的时间复杂度为 O(n)。

    var findNumberIn2DArray = function(matrix, target) {
      // 边界条件不能忘,数组为null或者为空数组时直接返回false
      if(matrix === null || matrix.length === 0) return false
      // 取到行数和列数
      var n = matrix.length, m = matrix[0].length;
      // 选定初始位置为右上角
      var i = 0, j = m-1;
      // 终止条件是到左下角
      while(i < n && j >= 0) {
          if(matrix[i][j] === target) {
              return true;
          // 比目标值大,则向左推进
          }else if(matrix[i][j] > target) {
              j--;
          // 比目标值小,则向下推进
          }else if(matrix[i][j] < target) {
              i++;
          }
      }
      return false;
  };