题目描述
在一个 n * m 的二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。
请完成一个高效的函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
示例
现有矩阵 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]
]
给定 target = 5,返回 true。
给定 target = 20,返回 false。
思路
- 校验数组为空,直接
return false
- 因为二维从左到右递增,从上到下递增,所以如果目标值小于二维数组左上角或者大于二维数组右上角的数值,直接
return false
- 选择从
右上角
往左下角
遍历,因为这样索引处的值,肯定是大于等于左边的,比小于等于下面的(也可以选择从左下角
往右上角
) - 如果行坐标大于等于数组的行数,或者列坐标小于0,那就表示遍历结束都没找到,
return false
- 如果目标值等于当前索引处值,那么直接
return false
;如果大于,行左边+1
,跳出此次循环;否则列坐标-1
Code
public class Solution {
public boolean findNumberIn2DArray(int[][] matrix, int target) {
if (null == matrix || 0 == matrix.length || 0 == matrix[0].length) {
return false;
}
if (target < matrix[0][0] || target > matrix[matrix.length - 1][matrix[0].length - 1]) {
return false;
}
int i = 0;
int j = matrix[0].length - 1;
while (i < matrix.length && j >= 0) {
if (target == matrix[i][j]) {
return true;
}
if (target > matrix[i][j]) {
i++;
continue;
}
j--;
}
return false;
}
}