剑指 Offer 04. 二维数组中的查找 java代码

题目描述

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

分析

因为数组行列均有序，我们可以利用这个特性来缩小查找范围。用下面的网格来表示数组的行从row1到row2，且列从col1到col2范围的所有数据（每个方格对应一个数）。

上图中黄色位置为当前区域 的“中点”，也就是数组中的matrix[midrow][midcol]，其中midrow=(row1+row2)/2, midcol=(col1+col2)/2。那么target与“中点”大小关系有三种情况。

• target == matrix[midrow][midcol]
  即找到了target，直接返回true
• target > matrix[midrow][midcol]
  根据数组的特性可知，左上角范围的 matrix[i][j] < matrix[midrow][midcol],所以我们可以直接排除这部分，寻找范围缩小到蓝色框部分。将蓝框部分按照紫色虚线分割，在这两个部分进行相同的操作。
  
• target < matrix[midrow][midcol]
  和第二种情况一样，右下角范围的 matrix[i][j] > matrix[midrow][midcol]，所以直接排除这部分，寻找范围缩小到蓝色框部分。将蓝框部分按照紫色虚线分割，在这两个部分进行相同的操作。

java代码

	public boolean findNumberIn2DArray(int[][] matrix, int target) {
        if(matrix.length == 0) return false;
        int row = matrix.length, col = matrix[0].length;
        
        return binaryFind(matrix, target, 0, 0, row-1, col-1);
    }

	public boolean binaryFind(int[][] matrix, int target, int row1, int col1, int row2, int col2){
        if(row1>row2 || col1>col2) return false;
        
        int midrow = (row1+row2)/2, midcol = (col1+col2)/2;
        if(matrix[midrow][midcol] == target) return true;
        else if(matrix[midrow][midcol] > target){
            if(binaryFind(matrix,target,row1,col1,midrow-1,col2) || binaryFind(matrix,target,midrow,col1,row2,midcol-1))
                return true;
            return false;
            
        }else{
            if(binaryFind(matrix,target,midrow+1,col1,row2,col2) || binaryFind(matrix,target,row1,midcol+1,midrow,col2))
                return true;
            return false;
        }
    }