[LeetCode] 74. 搜索二维矩阵 ☆☆☆(二分查找)

Posted 反光的小鱼儿

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了[LeetCode] 74. 搜索二维矩阵 ☆☆☆(二分查找)相关的知识,希望对你有一定的参考价值。

描述

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

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

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

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

解析

每行有序,且前后行也有序。在最后一列中,找到第一个 >= target的元素,确定target可能在的行。再用简单二分,在这一行查找target即可。

代码

public static boolean searchMatrix(int[][] matrix, int target) {
        if (null == matrix || matrix.length <= 0 || matrix[0].length <= 0) {
            return false;
        }
        int length = matrix.length;
        int[] lastCol = new int[length];//最后一列
        for (int i = 0; i < length; i++) {
            lastCol[i] = matrix[i][matrix[0].length - 1];
        }
        int lastColIndex = firstLargeIndex(lastCol, target);
        if (lastColIndex == -1) {
            return false;
        }
        //简单二分
        int[] ponitRow = matrix[lastColIndex];//确定了仅可能在这一行
        int start = 0;
        int end = ponitRow.length - 1;
        while (start <= end) {
            int mid = (start + end) >>> 1;
            if (ponitRow[mid] == target) {
                return true;
            } else if (ponitRow[mid] < target) {
                start = mid + 1;
            } else {
                end  = mid - 1;
            }
        }
        return false;
    }

    /**
     * 找第一个 >=target的元素位置
     * @param nums
     * @param target
     * @return
     */
    public static int firstLargeIndex(int[] nums, int target) {
        int start = 0;
        int end = nums.length - 1;
        while (start <= end) {
            int mid = (start + end) >>> 1;
            if (nums[mid] < target) {
                start = mid + 1;
            } else {
                end = mid - 1;
            }
        }
        if (start >= 0 && start <= nums.length - 1) {
            return start;
        }
        return -1;
    }

 

以上是关于[LeetCode] 74. 搜索二维矩阵 ☆☆☆(二分查找)的主要内容,如果未能解决你的问题,请参考以下文章

[JavaScript 刷题] 二分搜索 - 搜索二维矩阵, leetcode 74

[leetcode] 74. 搜索二维矩阵

Leetcode 74.搜索二维矩阵

leetcode(74)----搜索二维矩阵(二分查找)

LeetCode 74.搜索二维矩阵

leetcode-74-搜索二维矩阵