LeetCode 剑指Offer学习计划第五天(04.二维数组中的查找)

题目:

在一个 n * m 的二维数组中,每一行都按照从左到右 非递减 的顺序排序,每一列都按照从上到下 非递减 的顺序排序。请完成一个高效的函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。

示例:

现有矩阵 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。

一开始看到题目想到了二分法,首先对矩阵的第一行进行二分,找到包含target的最小区间(如上述示例,target=5时,最小区间即为matrix[0][1]-matrix[0][2]),然后再对这两列进行二分法,找到target,若找不到则返回false,其时间复杂度为(logn)*(logm),但鄙人数学基础以及编程基础比较渣,未能实现。

后来看了官方题解,里面有用二分法的方法,但它是对矩阵第一行整行进行遍历,然后找到最小区间,再对区间的列进行二分,其时间复杂度为m*(logn)。

然后又看到了一个大佬的解题步骤,其时间复杂度仅为m+n,他是从矩阵最后一行的第一个元素开始遍历,即matrix[n-1][0],若该元素大于target,则前往上一行,直至某一行k的元素小于target时,此时读取同一行的下一列元素,即matrix[k][1],重复上述操作,即遇到比target大的元素时,行序号减1,遇到比target小的元素时,列序号减1。

代码实现:

 bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) {
        int i = matrix.size() - 1, j = 0;
        while(i >= 0 && j < matrix[0].size())
        {
            if(matrix[i][j] > target) i--;
            else if(matrix[i][j] < target) j++;
            else return true;
        }
        return false;
    }

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值