【Leetcode050】二维数组中的查找/搜索二维矩阵 II

剑指 Offer 04. 二维数组中的查找【中等】

240、 搜索二维矩阵 II 【中等】

在一个 n * m 的二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。请完成一个高效的函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。

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

限制：
0 <= n <= 1000
0 <= m <= 1000

方法一：线性查找

1.1 思路分析

暴力法就不做了。可以根据第一行和第一列的数定位目标；
以第一行为基准，从第一行最后一个数开始与目标数进行比较，如果大于target，就往前找，如果小于就往下找，col+1，如果matrix[row][col] > target，就从此处往前找，直到找完还是没有就返回false。思路简单，但是实现起来有点绕，注意细节。

1.2 代码实现

class Solution {
    public boolean findNumberIn2DArray(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 row=0, col=cols-1;
        while(row < rows && col>=0){
            int num = matrix[row][col];
            if(num == target){
                return true;
            }else if(num < target){    // 往下找
                row++;
            }else{          // 往前找
                col--;
            }
        }
        return false;
    }      
}

1.3 测试结果

1.4 复杂度

• 时间复杂度：O(M+N)
• 空间复杂度：O(1)