# 240.二维数组中的查找

在一个 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 <= m <= 1000

代码:

右上角的值是当前行最大值,当前列最小值,如果目标值小于右上角值,那么只能向左搜索,如果大于右上角值,只能向下搜索,每次可以排除一行或者一列。 最坏需要找m+n次,时间复杂度O(m+n);

var searchMatrix = function (matrix, target) {
    // 边界校验
    if (matrix === null || !matrix.length || !matrix[0].length) return false;
    // m * n
    const rows = matrix.length;
    const columns = matrix[0].length;
    // 从右上角开始,因为他是边界值,即他的左侧较小,下一行的值较大
    let x = 0;
    let y = columns - 1;
    while (x < rows && y >= 0) {
        const cur = matrix[x][y];
        if (cur === target) return true;
        if (cur > target) {
            // 左移
            y--;
        } else {
            // 下移
            x++;
        }
    }
    return false;
};

# 写在最后

本篇是剑指Offer的第二题,俗话说好的合理的数据结构+算法才是写好代码的关键,不妨跟我一起来吧~

热门开源项目

上次更新于: 3/23/2022, 10:11:04 AM