leetcode240 剑指offer04 二维数组中的查找 Java

快速浏览：只看3、4就行

0 高效解法

1 题目描述

2 分析题目

第一反应是双重循环，但是效率很低。
然后试了一下dfs，超出时间限制。(ˉ▽ˉ；)…

发现题目中的关键点在于：

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

因此我们要利用这个特性来达成高效的算法

3 解法思路

• 缩小查找范围：
  • target行首 <= target <= target行尾
• 从第一行分别将行首和行尾和target进行比较查找满足范围条件的行，每找到一行则遍历查找这一行的所有数字。

注意：如果数组为空（两种空）则直接返回false，否则越界。

4 Java代码片

class Solution {
    public boolean findNumberIn2DArray(int[][] matrix, int target) {
        if(matrix.length == 0 || matrix[0].length == 0) return false;
        int row = matrix.length - 1;
        int col = matrix[0].length - 1;
        for(int i = 0; i <= row; i++){
            if(matrix[i][0] <= target && matrix[i][col] >= target){
                for(int j = 0; j <= col; j++){
                    if(matrix[i][j] == target) return true;
                }
            }
        }
        return false;
    }
}

5 复杂度

• 时间复杂度O(m+n)
• 空间复杂度O(1)

6 改进

可以将行内的遍历查找改为二分查找，时间复杂度可减至O(m+logn)