74. 搜索二维矩阵

搜索二维矩阵

编写一个高效的算法来判断 m x n 矩阵中，是否存在一个目标值。该矩阵具有如下特性：

• 每行中的整数从左到右按升序排列。
• 每行的第一个整数大于前一行的最后一个整数。

示例 1:

输入:
matrix = [
  [1,   3,  5,  7],
  [10, 11, 16, 20],
  [23, 30, 34, 50]
]
target = 3
输出: true

思路+代码+注释：

public class SevenFour {

    public static void main(String[] args) {
        System.out.print(searchMatrix(new int[][]{{1,3}},3));
    }

    public static boolean searchMatrix(int[][] matrix, int target) {
            /*
            思路：因为每一行的数据是有序的，只要确定target所在行，然后使用二分查找即可

            让target和每一行的第一个数进行比较，如果target小于第一行的第一个数那么该数在矩阵中不存在；如果target
            小于第x行的第一个数那么target就位于第x-1行，在x-1行采用二分查找进行搜索；这里有个特殊情况就是target位于最后一行，
            如果target大于最后一行第一个数那么在和最后一行最后一个数进行比较，如果比最后一个数大那么就不存在于最后一行，否则
            存在于最后一行
            如果target和某一行的第一个数相等那么存在
             */
            //特殊情况，数组为空
            if (matrix.length==0 || matrix[0].length==0)
            {
                return false;
            }
            //target所在行数
            int row=-1;
        for (int i = 0; i < matrix.length; i++) {
            if (target<matrix[i][0])
            {
                if (i==0)
                {
                    return false;
                }
                else {
                    row=i-1;
                    break;
                }
            }
            else if (target == matrix[i][0])
            {
                return true;
            }
            else
            {
                if (i==matrix.length-1)
                {
                    if (target <= matrix[i][matrix[0].length-1])
                    {
                        row=i;
                    }
                }
            }
        }
        //row为-1证明不存在目标数据
        if (row==-1)
        {
            return false;
        }
        //在第row行做二分查找
        return binarySearch(matrix[row],target);
    }

    static boolean binarySearch(int[] nums,int target)
    {
            int low=0;
            int high=nums.length-1;
            while (low<=high)
            {
                int mid=(low+high)/2;
                if (nums[mid] == target)
                {
                    return true;
                }else if (target < nums[mid])
                {
                    high=mid-1;
                }else {
                    low=mid+1;
                }
            }
            return false;
    }
}