【LeetCode热题100】【二分查找】搜索二维矩阵

2024-04-17 08:04:21 浏览数 (2)

题目链接:74. 搜索二维矩阵 - 力扣(LeetCode)

在一个有序二维数组里面查找元素,同【LeetCode热题100】【矩阵】搜索二维矩阵 II-CSDN博客

如果用二分查找,时间复杂度是log(mn),但是可以实现时间复杂度为O(m n)的,从右上角开始查找,如果当前元素比目标小,说明应该在下面范围,矩阵压扁,如果当前元素比目标大,说明在左边范围,矩阵压扁,这样最多只需要遍历m n个元素可以确定元素是否存在

代码语言:javascript复制
class Solution {
public:
    bool searchMatrix(vector<vector<int> > &matrix, int target) {
        int x = 0, y = matrix[0].size() - 1, n = matrix.size();
        while (x < n && y >= 0) {
            if (matrix[x][y] == target)
                return true;
            if (matrix[x][y] < target)
                  x;
            else
                --y;
        }
        return false;
    }
};

0 人点赞