题目描述
在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。
请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
样例
输入数组:
[
[1,2,8,9],
[2,4,9,12],
[4,7,10,13],
[6,8,11,15]
]
如果输入查找数值为7,则返回true,
如果输入查找数值为5,则返回false。
算法1
区间定位
首先,查对角线元素,因为对角线元素是每个正方形区域右下角的数字,是最大的
然后,查列
最后,查行
时间复杂度 $O(3n)$
C++ 代码
class Solution {
public:
bool searchArray(vector<vector<int>> array, int target) {
// ①先查对角线元素,定位区间
int i;
for(i = 0; i < array.size() && array[i][i] <= target; i++){
if(array[i][i] == target)
return true;
}
// 注:此处容易出现数组越界错误,一定要加上if判断
if(i < array.size()){
int k;
// ②先查列
for(k = i; k >= 0; k--){
if(array[i][k] == target)
return true;
}
// ③再查行
for(k = i; k >= 0; k--){
if(array[k][i] == target)
return true;
}
}
return false;
}
};