[leetcode-74]Search a 2D Matrix(c)

最新推荐文章于 2023-07-06 06:33:35 发布

zdavb

最新推荐文章于 2023-07-06 06:33:35 发布

阅读量365

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/zdavb/article/details/47298705

leetcode 专栏收录该内容

243 篇文章

订阅专栏

本文介绍了一种在具有特定排列规则的矩阵中搜索目标值的高效算法。通过利用矩阵的特性，实现时间复杂度为 O(log(N*M)) 的二分查找，或者采用从右上角开始的遍历方式，达到 O(m+n) 的时间复杂度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题描述：
Write an efficient algorithm that searches for a value in an m x n matrix. This matrix has the following properties:

Integers in each row are sorted from left to right.
The first integer of each row is greater than the last integer of the previous row.
For example,

Consider the following matrix:

[
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
Given target = 3, return true.

分析：搜索矩阵，而且矩阵有很大特点。每行增大，而且下一行大于上一行的最后一个值。那么把矩阵放平那就是一个有序的数组。显然用二分查找很容易找到，时间复杂度为Ｏ（ｌｏｇ（Ｎ＊Ｍ））
其实这道题还有另外一种解法，不过时间复杂度为O(m+n)。根据特点：
每行递增，每列递增。那么从右上角开始出发，如代码2所示：

代码如下：4ms

bool searchMatrix(int** matrix, int matrixRowSize, int matrixColSize, int target) {
    int right = matrixRowSize*matrixColSize-1;
    int left = 0;

    while(left<=right){
        int mid = (left+right)/2;
        int row = mid/matrixColSize;
        int col = mid%matrixColSize;

        if(matrix[row][col]==target)
            return true;
        else if(matrix[row][col]<target)
            left = mid+1;
        else 
            right = mid-1;
    }
    return false;
}

代码2如下：4ms

bool searchMatrix(int** matrix, int matrixRowSize, int matrixColSize, int target) {
    int row = 0;
    int col = matrixColSize-1;

    while(row>=0&&row<matrixRowSize&&col>=0&&col<matrixColSize){
        int val = matrix[row][col];
        if(val==target)
            return true;
        else if(val>target)
            col--;
        else
            row++;
    }
    return false;
}