剑指Offer面试题4二维数组中的查找

题目

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

思路一

由于该数组每一行是升序，每一列是升序，（从左下角开始）：

• 如果matrix[i][j] < target,则可以排除第j列，j++；
• 如果matrix[i][j] > target,则可以排除第i行,i–；
• matrix[i][j] = target, 返回true;

class Solution {
    public boolean findNumberIn2DArray(int[][] matrix, int target) {
        //Step1:定义一个结果变量；
        boolean ret = false;

        //step2：判断异常
        if(matrix == null || matrix.length == 0) return ret;
        if(matrix[0] == null || matrix[0].length == 0) return ret;

        //Step3:定义辅助变量
        int n = matrix.length;
        int m = matrix[0].length;
        int x = n-1;//行
        int y = 0;//列

        //Step4:算法主体
        while(y < m && x >= 0){
            if(matrix[x][y] < target){
                y++;
            }else if(matrix[x][y] > target){
                x--;
            }else{
                ret = true;
                break;
            }
        }

        //step5:输出返回值
        return ret;
    }

}

思路二

逐行二分法：每一行都用二分法寻找是否存在target,针对第i行，left=0,right = m-1, middle = (left+right)/2,-

• 如果matrix[i][middle] < target,left++;
• 如果matrix[i][middle] > target,right–;
• 否则，ret = true

class Solution {
    public boolean findNumberIn2DArray(int[][] matrix, int target) {
        //Step1:定义结果变量
        boolean ret = false;

        //Step2:判断异常
        if(matrix == null || matrix.length == 0) return ret;
        if(matrix[0] == null || matrix[0].length == 0) return ret;

        //逐行使用二分法
        //Step3:定义辅助变量
        // int n = matrix.length;
        int middleIdx;

        //Step4:逐行二分法主体
        for(int i = 0; i < matrix.length; i++){
            int left = 0, right = matrix[0].length-1;
            while(left <= right)
            {
                middleIdx = (left + right) / 2;
                if(matrix[i][middleIdx] < target){
                    left++;
                }else if(matrix[i][middleIdx] > target){
                    right--;
                }else{
                    ret = true;
                    break;
                }
            }
            // if(ret == true) break;
        }
        return ret;
    }
}