NowCoder LeetCode
题目描述
在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
解题思路
要求时间复杂度 O(M + N),空间复杂度 O(1)。其中 M 为行数,N 为 列数。
该二维数组中的一个数,小于它的数一定在其左边,大于它的数一定在其下边。因此,从右上角开始查找,就可以根据 target 和当前元素的大小关系来缩小查找区间,当前元素的查找区间为左下角的所有元素。
class Solution {
public:
bool Find(int target, vector<vector<int> > array) {
int rows = array.size(), columns = array[0].size();
if(!array.empty() && rows>0 && columns>0) {
int row = 0, column = columns-1;
while(row<rows && column>=0) {
if(array[row][column] == target) return true;
else if(array[row][column] > target) --column;
else ++row;
}
}
return false;
}
};