剑指 Offer 04. 二维数组中的查找(二分技巧)

24 篇文章 0 订阅
5 篇文章 0 订阅

在一个 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。 

【思路】:
此题用暴力做效率很低,虽然也发现了矩阵有序的特点,但不知如何更好的使用二分查找。

在题解中看到这样一句话深受启发:

站在右上角看。这个矩阵其实就像是一个Binary Search Tree

所以我突然反应出来,原来右上角是一个比较的起始位置。
如果目标比右上角元素大,则往下搜。如果目标比右上角元素小,则往左搜。可以用递归也可以用循环实现。


【代码】:

/**
 * @param {number[][]} matrix
 * @param {number} target
 * @return {boolean}
 */

var judge = function(row, col, matrix){
    if(row >= matrix.length || col >= matrix[0].length) return false;
    if(row < 0 || col < 0)  return false;
    return true;
}

var findNumberIn2DArray = function(matrix, target) {
    if(matrix.length == 0)  return 0;
    //从右上角开始
    let col = matrix[0].length - 1;
    let row = 0; 
	
	//如果出界了还没搜到那肯定不存在该元素
    while(judge(row, col, matrix)){ 
        if(matrix[row][col] < target){
            row++;
        }else if(matrix[row][col] > target){
            col--;
        }else{
            return true;
        }
    }
    return false;
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值