在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。输入一个二维数组和一个目标数,返回 false OR true。
暴力法
直接遍历,两层循环,找到后直接返回
public boolean findOn2(int[][] matrix, int number) {
boolean found = false;
if (matrix != null && matrix.length > 0)
for (int i = 0; i < matrix.length; i++)
for (int j = 0; j < matrix[i].length; j++)
if (matrix[i][j] == number) {
found = true;
break;
}
return found;
}
时间复杂度: 根据二维数组的大小
空间复杂度: O(1)
暴力法忽视了有序二维数组的条件。我们查找规律 ,发现可以首先选取数组中右上角的数字。如果该数据等于要查找的数字,则查找过程结束;如果该数字大于要查找的数字,则剔除这个数字所在的列:如果该数字小于要查找的数字,则剔除这个数字所在的行。也就是说,如果要查找的数字不在数组的右上角,则每一次都在数组的查找范围中提出一行或者一列,这样每一步都可以缩小查找的范围,直到找到要查找的数字,或者查找范围为空。
public boolean findNumberIn2DArray(int[][] matrix, int target) {
boolean found = false;
if (matrix != null && matrix.length > 0) {
int row = 0;
// 获取第一行的个数即是二维数组的列数
int column = matrix[row].length - 1;
while (row < matrix.length && column >= 0) {
// 选取数组右上角的数字,如果等于要查找的number,查找过程结束
if (matrix[row][column] == target) {
found = true;
break;
} else
// 如果该数字大于要找的数字,剔除该数字所在的列
if (matrix[row][column] > target)
column--;
// 如果该数字小于要查找的数字,剔除所在行
else if (matrix[row][column] < target)
row++;
}
}
return found;
}
时间复杂度: O(n+m)
空间复杂度: O(1)
同样的我们也可以通过查找右下角的数字的方式。
测试用例:
@Test
public void FindTest1() {
int[][] arr = {{1, 2, 8, 9}, {2, 4, 9, 12}, {4, 7, 10, 13}, {6, 8, 11, 15}};
int[][] arr1 = {};
int[][] arr2 = {{}};
Find2DArray find2DArray = new Find2DArray();
Assert.assertTrue(find2DArray.findOn2(arr, 7));
Assert.assertFalse(find2DArray.findOn2(arr1, 0));
Assert.assertFalse(find2DArray.findOn2(arr2, 1));
}
@Test
public void FindTest2() {
int[][] arr = {{1, 2, 8, 9}, {2, 4, 9, 12}, {4, 7, 10, 13}, {6, 8, 11, 15}};
int[][] arr1 = {};
int[][] arr2 = {{}};
Find2DArray find2DArray = new Find2DArray();
Assert.assertTrue(find2DArray.findNumberIn2DArray(arr, 7));
Assert.assertFalse(find2DArray.findNumberIn2DArray(arr1, 0));
Assert.assertFalse(find2DArray.findNumberIn2DArray(arr2, 1));
}