二维数组中的查找

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

思路解析:对于这个题,我们可能第一时间想到的就是使用暴力遍历整个矩阵,这个办法是可行的,但是此方法似乎没有用到矩阵从左到右,从上到下的特点,显然不是最优解法,时间复杂度达到了O(NM)。回想我们在一维递增数组中查找元素的时候,也是通过二分查找来提高效率的,不会从头到尾的遍历。那么对于这个二维数组,我们也需要利用递增这个特点,但是此时又有一个问题,我们需要从矩阵的哪个角开始遍历呢?如果是选择左上角和右下角,那么我们每判断一个元素,都需要从两个方向进行查找,这样就大大增加了题目的难度,如果选择左下角和右上角,我们则只需要控制一个方向的判断,它最终的时间复杂度只有O(N+M)。

我们以左下角的元素为起点,暂定为temp,若temp>target,那么target一定在temp所在行的上方;若temp<target,那么target一定在temp所在行的右边。根据这个特点,我们可以很快判断此元素是否在数组中。

图解:假定target=5;

 代码实现:

 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;
    }

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值