二维数组中的查找

  在一维数组中查找一个数有很多办法,比如最简单的暴力查找,或者排序之后在利用二分查找,如果涉及到二维数组,其实原理还是一样的。

在一个 n * m 的二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成
一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
[
  [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。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/er-wei-shu-zu-zhong-de-cha-zhao-lcof
  • 看到这个我们也可以用暴力搜索,也可以用二分查找,使用二分查找的时候就是一行一行的去查找;
class Solution {
    public boolean findNumberIn2DArray(int[][] matrix, int target) {
        if(matrix.length == 0){
            return false;
        }
        if(matrix[0].length == 0){
            return false;
        }
        int n = matrix.length;
        int m = matrix[0].length;
        if(target < matrix[0][0] || target > matrix[n-1][m-1]){
            return false;
        }
        int mid = -1;
        for(int i=0;i<n;i++){
            mid = binarySearch(matrix[i],target);
            if(mid != -1){
                return true;
            }
        }
        return  false;
    }

    //二分查找
    public int binarySearch(int[] arr,int target){
        int start = 0;
        int end = arr.length-1;
        int mid;
        while(start <= end){
            mid = start + (end - start) / 2;
            if(arr[mid] == target){
                return mid;
            }else if(arr[mid] > target){
                end = mid - 1;
            }else{
                start = mid + 1;
            }
        }
        return -1;
    }
}
  • 但是这里我们其实可以很好的发现是个递增数组,每一行每一列都是递增的,我们可以巧妙利用这个性质;此时就要求我们关注到矩阵的4个顶点了;
    思路
class Solution {
    public boolean findNumberIn2DArray(int[][] matrix, int target) {
        int row = matrix.length - 1;
        int col = 0;
        while(row >= 0 && col < matrix[0].length){
            if(target < matrix[row][col]) row--;
            else if(target > matrix[row][col]) col++;
            else if(target == matrix[row][col]) return true;
        }
        return false;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值