LeetCode 240. 搜索二维矩阵 II

注：该篇文章会与我的个人博客同步更新。欢迎移步https://cqh-i.github.io/体验更好的阅读效果。

题目描述

编写一个高效的算法来搜索$m\ast n$矩阵 matrix 中的一个目标值 target。该矩阵具有以下特性：

• 每行的元素从左到右升序排列。
• 每列的元素从上到下升序排列。

示例:

现有矩阵 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。

思路

这道题与之前的LeetCode 74. 搜索二维矩阵有些类似, 但是矩阵的特性有点不一样. 每列的元素从上到下升序排列,给使用二分查找造成了麻烦. 一开始我想将利用二分查找将矩阵分成四块,然后递归地去查找,但是写起来有点复杂,如下:

假设我们第一个取的元素是17,如果要查找的值小于17,那么就要去递归查找左上角,右上角以及左下角的矩阵; 如果取的元素是9,要查找的值大于9,那么就要递归查找右上角,左下角以及右下角的矩阵.复杂度高,放弃这种做法.
其实还有左上角和右上角两个元素是值得注意的,以左下角的元素为例,矩阵的第一列和最后一行连起来就是一个有序的数组.

如果我们要查找的元素小于18,那么就可以抛弃矩阵的最后一行,如果我们要查找的元素大于18,那么就可以抛弃矩阵的第一列,这样就把问题的规模不断缩小.时间复杂度为
O(max(num_row, num_column))

class Solution {
    public boolean searchMatrix(int[][] matrix, int target) {
        if (matrix == null || matrix.length == 0 || matrix[0].length == 0)
            return false;
        int num_row = matrix.length;
        int num_column = matrix[0].length;
        int begin_row = num_row - 1;
        int begin_column = 0;
        while (begin_row >= 0 && begin_column < num_column) {
            if (matrix[begin_row][begin_column] == target)
                return true;
            if (matrix[begin_row][begin_column] > target) {
                --begin_row;
            } else {
                ++begin_column;
            }
        }
        return false;
    }
}