题目:在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
Consider the following matrix:
[
[1, 4, 7, 11, 15],
[2, 5, 8, 12, 19],
[3, 6, 9, 16, 22],
[10, 13, 14, 17, 24],
[18, 21, 23, 26, 30]
]Given target = 5, return true.
Given target = 20, return false.
从题目中获取有效信息:
- 数组中元素右边的数一定大于左边的数;
- 数组中元素下面的数一定大于上面的数。
通过这个规律可以找一个临界的位置,逐渐移动查找坐标来缩小范围,这样就不用对数组中每个元素做遍历比较了。
有两种范围缩小方式:
- 每次减少一列(当前遍历元素大于目标值)或者每次增加一行(当前遍历元素小于目标值);
- 每次增加一列(当前遍历元素小于目标值)或者每次减少一行(当前遍历元素大于目标值)。
通过两种不同的范围缩小方式我们很容易确定遍历的开始位置
第一种方式:右上角元素;
第二种方式:左下角元素。
从右上角元素开始遍历,代码示例如下所示:
public boolean Find(int target, int [][] array) {
// 二维数组非空校验
if (null == array
|| 0 == array.length
|| null == array[0]
|| 0 == array[0].length) {
return false;
}
// 获取行和列
int row = array.length - 1;
int col = array[0].length - 1;
// 从右上开始遍历
int i = 0;
int j = col;
// 注意边界值,不能超过行数并且列数不能比0小
while (i <= row && j >= 0) {
if (array[i][j] < target) {
i++;
} else if (array[i][j] > target) {
j--;
} else {
return true;
}
}
return false;
}
- 若行数为m,列数为n,时间复杂度由行和列数两个数据规模决定,所以渐进时间复杂度为O(m+n)
- 空间复杂度为O(1)