剑指Offer-二维数组中的查找

发表于 | 分类于

题目描述

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

image.png

解题思路

参考github: https://github.com/CyC2018/CS-Notes

一开始的想法就是很直接的，直接一行行的查找，因此时间复杂度也相对会大一些，尽管也能够通过，但却是最笨的方法。下面写写时间复杂度为O(M+N)+O(1)的写法。

从右上角开始查找。矩阵中的一个数，它左边的数都比它小，下边的数都比它大。因此，从右上角开始查找，就可以根据 target 和当前元素的大小关系来缩小查找区间。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
class Solution {
public:
    /*bool Find(int target, vector<vector<int> > array) {
        for(int i=0; i<array.size(); i++){
            for(int j=0; j<array[0].size(); j++){
                if(array[i][j] == target) return 1;
            }
        }
        return 0;
    }*/
    
    bool Find(int target, vector<vector<int>> array){
        int rows = array.size(), cols = array[0].size();
        int r = 0, c = cols-1;
        while(r <= rows-1 && c >= 0){
            if(target == array[r][c]) return 1;
            else if(target > array[r][c]) r++;
            else c--;
        }
        return 0;
    }
};