【力扣】- 剑指 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。

限制:

0 <= n <= 1000

0 <= m <= 1000

题目链接

解法:
第一种方法直接遍历数组找,第二种对每一行进行二分。
第三种o(n+m)做法,即我们观察可以发现当我们位于右上角时,向下走会使得结果变大,向左走会使得结果变小。当我们位于左下角时,向上走会使得结果变小,向右走会使得结果增大。那么我们只需要选取这两个点其中一个,然后根据当前值与target的大小关系选择对应的方向即可。

对每一行进行二分:

扫描二维码关注公众号,回复: 12725812 查看本文章
class Solution {
    
    
    public boolean findNumberIn2DArray(int[][] matrix, int target) {
    
    
        for(int i=0;i<matrix.length;i++){
    
    
            int l=0,r=matrix[i].length-1;
            while(l<=r){
    
    
                int mid=(l+r)/2;
                if(matrix[i][mid]==target){
    
    
                    return true;
                }
                if(matrix[i][mid]<target){
    
    
                    l=mid+1;
                }else{
    
    
                    r=mid-1;
                }
            }
        }
        return false;
    }
}

线性解法:

class Solution {
    
    
    public boolean findNumberIn2DArray(int[][] matrix, int target) {
    
    
        if(matrix.length==0){
    
    
            return false;  //防止下标越界异常
        } 
        int n=matrix.length; //行数
        int m=matrix[0].length; //列数
        int row=0,col=m-1;
        while(row<n&&col>=0){
    
    
            if(matrix[row][col]==target){
    
    
                return true;
            }
            if(matrix[row][col]<target){
    
    
                row++;
            }else{
    
    
                col--;
            }
        }
        return false;
    }
}

猜你喜欢

转载自blog.csdn.net/tran_sient/article/details/114398097