题目描述
在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
第一种方法:
遍历每个一维数组(当一维数组的第一个元素大于所给整数时则停止),对于每个一维数组进行二分查找。复杂度为
O
(
m
l
o
g
n
)
O(mlogn)
O(mlogn)
bool Find(int target, vector<vector<int> > array) {
for(int i = 0; i < array.size() && array[i][0] < target;i++)
{
int l = 0, r = array[i].size()-1;
while(l <= r)
{
int mid = (l + r)/2;
if(array[i][mid] > target)
r = mid - 1;
else if(array[i][mid] < target)
l = mid + 1;
else return true;
}
}
return false;
}
这个在牛客网上报错段错误,但是把for语句中的&& array[i][0] < target
去掉就过了…,但是感觉这句不存在越界啊。。。,有大佬知道原因请评论指出,谢谢。
第二种方法:
由题可知二维数组中左下角的数最小,用这个数与查找的整数进行比较,若小于查找的整数则向右走,若大于则向上走,等于则证明找到了。若超出边界还没找到则二维数组中没有该数。时间复杂度为
O
(
m
+
n
)
O(m+n)
O(m+n)
bool Find(int target, vector<vector<int> > array) {
int i = array.size()-1;
int r = array[0].size();
int j = 0;
while(i >= 0 && j < r)
{
if(array[i][j] > target)
i--;
else if(array[i][j] < target)
j++;
else return true;
}
return false;
}