【题目】
在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
【思路】
比较目标数字target和二维数组右上角数字的大小:
-
如果target大于这个数字,由于二维数组右上角数字是这一行最大的数,所以target不可能在这一行,剔除该行。
-
如果target小于这个数字,由于二维数组右上角数字是这一列最小的数,所以target不可能在这一列,剔除该列。
-
如果target等于这个数字,说明找到了,返回真。循环结束还没找到,说明数组中不含有target,返回假。
【代码】
class Solution {
public:
bool Find(int target, vector<vector<int> > array) {
int rows = array.size(); //行数
int cols = array[0].size(); //列数
//没有下面这个判断答案也通过,但加上比较严谨
if (rows == 0 || cols == 0) return false;
int row = 0;
int col = cols - 1;
while (row < rows && col >= 0) {
if (target > array[row][col]) {
row++;
}
else if (target < array[row][col]) {
col--;
}
else { //等于
return true;
}
}
return false;
}
};