牛客JZ1 二维数组中的查找
在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
[1,2,8,9],
[2,4,9,12],
[4,7,10,13],
[6,8,11,15]给定 target = 7,返回 true。
给定 target = 3,返回 false。
思路:因为每一行和每一列都是从上到下递增的顺序,所以可以从左下角或者右上角出发搜索,每次可以排除一行或一列。
class Solution {
public:
bool Find(int target, vector > array) {
// 从左下角开始搜索
int row = array.size() - 1, col = 0;
while (row >= 0 && col < array[0].size()) {
// 小于target列向右移;大于target则行向上移
if (array[row][col] < target) {
col++;
}
else if (array[row][col] > target) {
row--;
}
else {
return true;
}
}
return false;
}
};