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

74. 搜索二维矩阵

240. 搜索二维矩阵 II

题目描述

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

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

输入输出

1
2
3
4
5
输入：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

输入：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 = 20
输出：false

基本思路

  • 暴力搜索 时复O(mn) 面试直接寄
  • 线性查找：从左下角或者右上角开始搜索 时复$O(m+n)$ 空复$O(1)$

java实现

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
class Solution {
    public boolean searchMatrix(int[][] matrix, int target) {
        if(matrix == null || matrix.length == 0 || matrix[0].length == 0){
            return false;
        }
        int rows = matrix.length, columns = matrix[0].length;
        int row = 0, column = columns - 1;
        while(row < rows && column >= 0){
            int num = matrix[row][column];
            if(num == target){
                return true;
            }
            else if(num > target){
                column--;
            }
            else{
                row++;
            }
        }
        return false;
    }
}