题目描述:
在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
解题思路:
由于元素有一定的顺序特点,可以从左下角或者右上角的元素开始与target进行比较。这里我选择从左下角开始,当target < a[row][col]时,那么target一定出现在上一行,即–row;当 target > a[row][col]时,那么target一定出现在右边,即++col;当target == a[row][col]时,即返回true。边界条件是行不能小于最上边的行,列不能大于最右边的一列,即row>=0 && col<col_max。
代码如下:
C++版本:
class Solution {
public:
bool Find(int target, vector<vector<int> > array) {
if(array.empty() || array[0].empty())
return false;
int row_max = array.size();
int col_max = array[0].size();
int row = row_max - 1;
int col = 0;
while(row >= 0 && col < col_max)
{
if(target < array[row][col])
--row;
else if(target > array[row][col])
++col;
else
return true;
}
return false;
}
};
Java版本:
public class Solution {
public boolean Find(int target, int [][] array) {
int row_max = array.length;
int col_max = array[0].length;
int row = row_max-1;
int col = 0;
while(row>=0 && col<col_max)
{
if(target == array[row][col])
return true;
else if(target > array[row][col])
col++;
else
row--;
}
return false;
}
}