[二分查找 ]74. 搜索二维矩阵（二维数组上的二分查找）

第74题.搜索二维矩阵（二分查找）

74. 搜索二维矩阵

分类：二维数组、二分查找

思路1：先二分找到所在行，再二分查找所在列

先取矩阵每一行的第0个元素做二分查找，找到target可能的行，确定了所在行后，对这一行内部再做二分查找，找到所在的列。

根据矩阵的特点，先取矩阵每一行的第1个元素作为二分的区间，例如：

  [1,   3,  5,  7],
  [10, 11, 16, 20],
  [23, 30, 34, 50]，target=3

取left=0表示第0行，right=matrix.length-1表示最后一行，mid=(left+right)/2，
先拿第mid行的第0个元素和target比较：

• 如果=target，直接返回true
• 如果<target，说明target可能在mid行和right行之间，为了缩小查找区间，再取mid行的最后一个元素和target比较，如果<target，则target可能在mid+1和right之间，如果>target，则target只可能在mid行。（关键点）
• 如果>target，说明target可能在left行和mid-1行之间；

本例中target=3，按二分查找的思想继缩小查找区间为[left,right=mid-1]，
此时left=rigth，所以开始对第left行内部做二分查找，退化成普通的二分查找。

实现遇到的问题：

1、特殊用例的处理（易忽略）

[], [[]], null
分别对应：

matrix.length == 0,
matirx[0].length == 0
matrix == null

2、代码实现需要写两个二分查找代码

寻找可能的行需要优化见思路2，只需要一个二分查找代码。

实现代码

class Solution {
    public boolean searchMatrix(int[][] matrix, int target) {
        if(matrix == null || matrix.length == 0 || matrix[0].length == 0) return false;
        int rows = matrix.length, cols = matrix[0].length;
        int left = 0, right = rows - 1;
        while(left <= right){
            //left==right时说明target只可能在这一行之内，对这一行做二分查找：普通的二分查找
            if(left == right){
                int start = 0, end = cols - 1;
                while(start < end){//start==end时退出循环
                    int middle = (start + end) / 2;
                    if(matrix[left][middle] == target) return true;
                    else if(matrix[left][middle] > target) end = middle - 1;
                    else if(matrix[left][middle] < target) start = middle + 1;
                }
                //start==end时退出循环，所以start==end的元素没有得到检验，最后需要单独做一次检验
                return matrix[left][start] == target;
            }
            //二分寻找target可能存在的行
            else{
                int mid = (left + right) / 2;
                if(matrix[mid][0] == target) return true;
                //matrix[mid][0]>target，说明target可能在left行和mid-1行之间；
                else if(matrix[mid][0] > target){
                    right = mid - 1;
                }
                //matrix[mid][0]<target，target有可能在mid行上，或在[mid+1,right]行之内，需要取mid行最后一个元素再做一次判断，进一步缩小范围
                else if(matrix[mid][0] < target){
                    if(matrix[mid][cols - 1] == target) return true;
                    else if(matrix[mid][cols - 1] < target) left = mid + 1;
                    else if(matrix[mid][cols - 1] > target){//target只可能在mid行
                        left = mid;
                        right = mid;
                    }
                }
            }
        }
        return false;
    }
}

思路2：将二维下标转换为一维下标，相当于在一维数组上做二分查找（推荐）

二维数组下标转一维数组：

例如：
  [1,   3,  5,  7],
  [10, 11, 16, 20],
  [23, 30, 34, 50]
转成：
  [1,   3,  5,  7, 10, 11, 16, 20, 23, 30, 34, 50]

下标关系式为：设(i,j)表示二维数组坐标，idx表示转换后的一维数组下标

idx = i * cols + j，i = idx / cols, j = idx % cols

初始时一维数组的左右边界left=0, right = rows * cols - 1，

使用上述公式可以将二维数组模拟成一维数组进行二分查找。

实现遇到的问题：

特殊用例的处理（易忽略）

[], [[]], null
分别对应：

matrix.length == 0,
matirx[0].length == 0
matrix == null

实现代码

class Solution {
    public boolean searchMatrix(int[][] matrix, int target) {
        //
        if(matrix == null || matrix.length == 0 || matrix[0].length == 0) return false;
        int rows = matrix.length, cols = matrix[0].length;
        int left = 0, right = rows * cols - 1;
        while(left <= right){
            int mid = (left + right) / 2;//一维数组下标
            //将一维数组下标转成二维数组
            int i = mid / cols, j = mid % cols;
            if(matrix[i][j] == target) return true;
            else if(matrix[i][j] < target) left = mid + 1;
            else if(matrix[i][j] > target) right = mid - 1;
        }
        return false;
    }
}