在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
很久以前就做过这道题,那会儿刚刷题也没想怎么优化,前几天室友面试的回来分享了这道面试题,又来看看,又一次证明自己太懒了,N^2的复杂度以前也能接收。这里又重新做了一遍,见代码:
class Solution {
public:
bool search(vector<vector<int> > array, int rows, int cols, int target)
{
int i = 0, j = cols - 1;
while (i < rows && j >= 0)
{
if (array[i][j] == target)
{
return true;
}
else if (array[i][j] > target)
{
j--;
}
else
{
i++;
}
}
return false;
}
bool Find(int target, vector<vector<int> > array) {
int rows = array.size();
int cols = array[0].size();
return search(array, rows, cols, target);
}
};