二维数组的查找
题目描述
在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
暴力搜索
拿到题目,就想到了直接暴力搜索,由于每一行、每一列是递增的顺序,就可以暴力从行开始搜索。
时间复杂度O(n2),空间复杂度O(1)。
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[i].size(); ++j) {
if (target == array[i][j])
{
return true;
}
}
}
return false;
}
};
规律搜索
每一行从左往右递增,每一列从下往上递增。左下角的数是m行,n列中最小的数。
搜索过程:以左下角的数为基准
- 如果比target的小,列数增加再重复过程
- 如果比target的大,行数减小再重复过程
- 如果与target一样大,返回true
复杂度分析
时间复杂度:O(n+m)。在循环语句中,除非直接返回结果否则每一次行会减少或者列增加一次,矩阵共有m行n列,因此在循环截止之前会执行至多m+n次。
空间复杂度:O(1)。没有额外的存储空间。
class Solution {
public:
bool Find(int target, vector<vector<int> > array) {
int rows = array.size();
int cols = array[0].size();
int row = rows - 1;
int col = 0;
while (row >= 0 && col < cols)
{
if (array[row][col] > target)
{
row--;
}
else if (array[row][col] == target)
{
return true;
}
else
{
col++;
}
}
return false;
}
};