该二维数组特殊处是行、列都为增序排列。
方法一:
暴力解法,遍历数组查找该值:
public boolean findNumberIn2DArray(int[][] matrix, int target) {
if (matrix == null || matrix.length == 0 || matrix[0].length == 0) {
return false;
}
int rows = matrix.length, columns = matrix[0].length;
for (int i = 0; i < rows; i++) {
for (int j = 0; j < columns; j++) {
if (matrix[i][j] == target) {
return true;
}
}
}
return false;
}
方法二:利用set或list的contain
将所有数据加入set,再用contain方法判断是否存在。
public boolean findNumberIn2DArray1(int[][] matrix, int target) {
Set<Integer> nums = new HashSet<>();
for (int i = 0; i < matrix.length; i++) {
for (int j = 0; j < matrix[0].length; j++) {
nums.add(matrix[i][j]);
}
}
return nums.contains(target);
}
方法三:转换看待该二维矩阵的思维:
正常看待二维矩阵是看做一个长方形,但该题是行、列都为增序,那么当我们将二维矩阵逆时针旋转45°,获得一个菱形,该菱形向左下走都是递减,向右下走都是递增,并且向左下走为列数-1,向右下走位行数+1,这样从最大列数、最小行数进行遍历即可找到该数。
画出图可看似二叉树查找。
public boolean findNumberIn2DArray2(int[][] matrix, int target) {
int i = matrix.length - 1, j = 0;
while(i >= 0 && j < matrix[0].length)
{
if(matrix[i][j] > target) {
i--;
} else if(matrix[i][j] < target) {
j++;
} else {
return true;
}
}
return false;
}
方法四:将二维矩阵看作坐标系:
类似方法三,都是在找这个特殊二维矩阵数值布局规律,带入坐标系的思路,将其左下角看作原点,y轴是递减序列,x轴是递增,接着按该思路遍历即可。
注:java中获取二维数组的行列,matrix.length是行数,matrix[0].length是列数
public boolean findNumberIn2DArray3(int[][] matrix, int target) {
int rows = matrix.length;
int columns = matrix[0].length;
int row = 0;
int column = columns - 1;
while (row < rows && column >= 0) {
int num = matrix[row][column];
if (num == target) {
return true;
} else if (num > target) {
column--;
} else {
row++;
}
}
return false;
}