【算法】搜索二维矩阵

难度:中等;

题目:

给你一个满足下述两条属性的 m x n 整数矩阵:

  • 每行中的整数从左到右按非严格递增顺序排列。
  • 每行的第一个整数大于前一行的最后一个整数。

给你一个整数 target ,如果 target 在矩阵中,返回 true ;否则,返回 false

示例 1:

lua 复制代码
输入:matrix = [[1,3,5,7],[10,11,16,20],[23,30,34,60]], target = 3
输出:true

示例 2:

lua 复制代码
输入:matrix = [[1,3,5,7],[10,11,16,20],[23,30,34,60]], target = 13
输出:false

提示:

  • m == matrix.length
  • n == matrix[i].length
  • 1 <= m, n <= 100
  • -104 <= matrix[i][j], target <= 104

解题思路:

可以通过使用二分查找的思想在一个二维矩阵中查找目标值。由于矩阵的每一行和每一列都是非严格递增的,并且每行的第一个元素都大于前一行的最后一个元素,我们可以将矩阵视为一个一维有序数组的组合。这就允许我们首先在行上应用二分查找,然后在选定的行上再次应用二分查找来定位目标值。

  1. 初始化 :定义两个指针 rowStartrowEnd,分别初始化为矩阵的第一行(0)和最后一行(matrix.length - 1)。
  2. 在行上二分查找 :在 rowStartrowEnd 之间进行二分查找,以找到可能包含目标值的那一行。如果 target 大于中间行的首个元素并且小于等于中间行的末尾元素,或者 target 大于中间行的下一个行的首个元素(如果存在),则在该行中查找 target
  3. 在行内二分查找 :如果找到了可能包含目标值的行,对该行使用二分查找来确定 target 是否存在于该行中。
  4. 返回结果 :如果在某一行中找到了 target,则返回 true;否则,如果所有行都未找到 target,则返回 false

JavaScript代码实现:

ini 复制代码
/**
 * @param {number[][]} matrix
 * @param {number} target
 * @return {boolean}
 */
var searchMatrix = function(matrix, target) {
    let rowStart = 0;
    let rowEnd = matrix.length - 1;
    let rowMid; // 声明 rowMid 变量

    // 二分查找合适的行
    while (rowStart <= rowEnd) {
        rowMid = Math.floor((rowStart + rowEnd) / 2); // 在循环内赋值
        if (target >= matrix[rowMid][0] && target <= matrix[rowMid][matrix[rowMid].length - 1]) {
            break;
        } else if (target < matrix[rowMid][0]) {
            rowEnd = rowMid - 1;
        } else {
            rowStart = rowMid + 1;
        }
    }

    // 如果没找到合适的行,直接返回 false
    if (rowStart > rowEnd) return false;

    // 二分查找行内的元素
    let colStart = 0;
    let colEnd = matrix[rowMid].length - 1;
    while (colStart <= colEnd) {
        let colMid = Math.floor((colStart + colEnd) / 2);
        if (matrix[rowMid][colMid] === target) {
            return true;
        } else if (matrix[rowMid][colMid] < target) {
            colStart = colMid + 1;
        } else {
            colEnd = colMid - 1;
        }
    }

    return false;
};

// 示例调用
// console.log(searchMatrix([[1, 3, 5, 7], [10, 11, 16, 20], [23, 30, 34, 50]], 3)); // 输出: true
// console.log(searchMatrix([[1, 3, 5, 7], [10, 11, 16, 20], [23, 30, 34, 50]], 13)); // 输出: false
相关推荐
DuHz3 小时前
论文精读:大语言模型 (Large Language Models, LLM) —— 一项调查
论文阅读·人工智能·深度学习·算法·机器学习·计算机视觉·语言模型
是烟花哈3 小时前
【前端】React框架学习
前端·学习·react.js
檀越剑指大厂3 小时前
32 万星的面试学习计划 + 内网穿透工具,程序员面试准备效率翻倍!
学习·面试·职场和发展
加农炮手Jinx4 小时前
LeetCode 72. Edit Distance 题解
算法·leetcode·力扣
offer收割机小鹅4 小时前
大学生求职必备:AI面试、AI写作与设计工具助力职场发展
人工智能·ai·面试·aigc·ai写作
精神阿祝4 小时前
“八股文”在程序员面试中的价值:助力还是阻力?
面试·职场和发展
借雨醉东风4 小时前
程序分享--常见算法/编程面试题:旋转矩阵
c++·线性代数·算法·面试·职场和发展·矩阵
逻辑驱动的ken4 小时前
Java高频面试考点场景题14
java·开发语言·深度学习·面试·职场和发展·求职招聘·春招
_深海凉_4 小时前
LeetCode热题100-打家劫舍
算法·leetcode·职场和发展
qq4356947014 小时前
JavaWeb08
前端