码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2020-02-07 01:13:28      阅读:61      评论:0      收藏:0      [点我收藏+]

标签:row   mat   有序   bsp   示例   bool   搜索   二分   存在   

描述

编写一个高效的算法来判断 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. 搜索二维矩阵 ☆☆☆(二分查找)

标签:row   mat   有序   bsp   示例   bool   搜索   二分   存在   

原文地址:https://www.cnblogs.com/fanguangdexiaoyuer/p/12271648.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!