剑指offer(1)二维数组中的查找

剑指offer(1)二维数组中的查找

题目

在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。

剑指offer(1)二维数组中的查找

分析

本题中数组从左到右递增,从上到下递增的数组,如果我们单纯使用两层遍历(时间复杂度为n2n^{2}显然是浪费了这一条件。有一种方式可以使用一层遍历就可以实现(时间复杂度为O(n))。当从右上角开始查找时,如果当前值比key值小,则向下走,否则向左走。同理也可以从左下角开始查找。但是不能从左上或右下开始查找。

代码


class Solution {
public:
    bool Find(int target, vector<vector<int> > array) {
        int rows=0;
        int cols=0;
        rows=array.size()-1;
        if(rows<0)
            return false;
        cols=array[0].size()-1;
        int i=0;
        int j=0;
       
        for(i=rows,j=0;i>=0&&j<=cols;)
        {
            if(array[i][j]==target)
                return true;
            else if(array[i][j] > target)
            {
                i--;
            }
            else
                j++;
        }
        return false;
    }
};

说明:

上述代码已通过牛客网ac。

参考文章:《剑指offer》

转载请附原文链接。