欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

(3)搜索二维矩阵 II

程序员文章站 2022-07-14 15:11:41
...

搜索二维矩阵 II

编写一个高效的算法来搜索 m x 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。

1、第一种解法(暴力搜索)

思路:双层循环进行遍历。
时间复杂度是o(n^2)。
代码:

public static boolean searchMatrix(int[][] matrix, int target) {
    for(int i=0; i < matrix.length; i++)
        for(int j=0; j < matrix[i].length; j++)
            if(target == matrix[i][j])
                return true;
    return false;
    }

2、第二种解法(利用最左下角元素的特性)

思路:可以发现,最左下角元素是当前列的最大元素,是当前行的最小元素。从最左下角元素开始,如果相等,返回true;如果target < matrix[i][j],则去掉当前行,问题规模可以减小为在去掉最后一行的子矩阵中寻找目标;如果target > matrix[i][j],则去掉当前列,问题规模可以减小为在去掉第一列的子矩阵中寻找目标。
时间复杂度o(n)。
代码:

private static boolean searchMatrix1(int[][] matrix, int target) {
    if(matrix.length == 0)
        return false;
    if(matrix[0].length == 0)
        return false;
    int i = matrix.length - 1;
    int j = 0;
    while(i >=0 && j < matrix[0].length){
        if(matrix[i][j] == target){
            System.out.println("("+i+","+j+")");
            return true;
        }

        else if(matrix[i][j] > target)
            --i;
        else
            ++j;
    }
    return false;
    }