一起来刷题备战系列


二维数组中的查找

https://leetcode-cn.com/problems/er-wei-shu-zu-zhong-de-cha-zhao-lcof/。
在一个 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]
]

二分查找

因为每一行都是从左到右有序的,所以可以对每个matrix[i]进行二分查找,可以初筛matrix[i][matrix[i].length-1] 如果比target要小,那么这一行都不可能满足条件,直接换下一行。

/**
 * @param {number[][]} matrix
 * @param {number} target
 * @return {boolean}
 */
var findNumberIn2DArray = function(matrix, target) {

    // 二分法
    dfs= (lis)=>{
        let sta = 0
        let end = lis.length-1
        let mid 
        // 在闭区间【a,b】中查找  即使a=b,也可以查一次,所以循环条件是<=
        while (sta<=end){
            mid = parseInt((sta+end)/2)
            if (lis[mid]===target) return true
            else if (lis[mid]>target) {
                end = mid-1
            }
            else if (lis[mid]<target){
                sta = mid+1
            }
        }

        return false 
    }

 

    for (let i=0;i<matrix.length;++i){
        let end = matrix[i].length-1
        if (matrix[i][end]>=target){
            let flag = dfs(matrix[i])
            if (flag) return true
        }
    }
    return false 

};

二叉搜索树思想,左小右大根在中间

将矩阵逆时针旋转45度,即可是一个二叉搜索树

class Solution:
    def findNumberIn2DArray(self, matrix: List[List[int]], target: int) -> bool:
        # 逆时针旋转45度
        n = len(matrix)
        if n==0:
            return False
        
        m = len(matrix[0])

        if  m  ==0 :
            return False
        cur = [0,m-1]
        x,y = cur
        while x<n and x>=0 and y>=0 and y<m:
            if matrix[x][y]>target:
                y-=1
            elif matrix[x][y]<target:
                x+=1
            else:
                return True

        return False 
        

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值