Leetcode NO.74 Search a 2D Matrix

目录

• 1.问题描述
• 2.测试用例
  • 示例1
  • 示例2
• 3.提示
• 4.代码

1.问题描述

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

• 每行中的整数从左到右按升序排列。

• 每行的第一个整数大于前一行的最后一个整数。

2.测试用例

示例1

示例2

3.提示

• m == matrix.length
• n == matrix[i].length
• 1 <= m, n <= 100
• -104 <= matrix[i][j], target <= 104

4.代码

二维矩阵降维成逻辑一维矩阵 & binary search

class Solution {
    public boolean searchMatrix(int[][] matrix, int target) {
        if (matrix == null || matrix.length < 1 || matrix[0].length < 1) {
            return false;
        }
        int m = matrix.length;
        int n = matrix[0].length;
        int l = 0;
        int r = m * n - 1;
        while (l <= r) {
            int mid = (l + r) / 2;
            int midValue = matrix[mid / n][mid % n];
            if (midValue == target) {
                return true;
            } else if (target < midValue) {
                r = mid - 1;
            } else {
                l = mid + 1;
            }
        }
        return false;
    }
}

时间复杂度 O(log(mn))
空间复杂度 O(1)