《剑指offer》刷题第一篇

【二维数组中的查找】原题连接https://www.nowcoder.com/practice/abc3fe2ce8e146608e868a70efebf62e?tpId=13&tqId=11154&tPage=1&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking
题目描述
在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。

class Solution {
public:
    bool Find(int target, vector<vector<int> > array) {
        for (int i = 0; i < (int)array.size(); i++)
            if (binary_search(array[i].begin(), array[i].end(), target))
                return true;
        return false;
    }
};

binary_search, 先来看一下这个函数的定义:
binary_search(first,last,value,comp)

first表示容器的头,last表示容器的尾,value表示想要查找的元素,comp比较函数(可以省略)

但是binary_search(二分查找)时间复杂度为 O(log n), 需要容器内数据呈现有序排列,升序或降序,返回的是一个布尔类型,只能告诉你这个这个容器中是否有这个数据,但不能确定其位置。(确定其位置需要用到 lower_band 函数)

方法二:从右上角开始查找,若大于target则删除该列(- -c),否则增加行(++r)。时间复杂度为 O(n)。

class Solution {
public:
    bool Find(int target, vector<vector<int> > array) {
        int n = array.size(), m = array[0].size();
        int r = 0, c = m - 1;
        while(r < m && c >= 0)
            if(array[r][c] == target)
                return true;
        else if(array[r][c] > target)
            --c;
        else
            ++r;
        return false;
    }
};

你可能感兴趣的:(剑指offer)