剑指Offer 04题
二维数组中的查找
-
题目描述:在一个 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
。 -
思路一:暴力法
即遍历数组的每一个元素进行比较。
-
思路二:二分查找
根据该数组的特征,从左下角或右下角元素开始比较,通过下标的增减来不断缩小范围。
public class Test04 {
public static void main(String[] args) {
int[][] array = {{1,2,8,9,16},{2,4,9,12,17},{4,7,10,13,18},{6,8,11,15,19}};
//System.out.println(array.length); //4
Test04 test = new Test04();
System.out.println(test.findNumberIn2DArray(array, 7));
System.out.println(test.findNumberIn2DArray2(array, 7));
}
//暴力法
public boolean findNumberIn2DArray(int[][] array, int target){
for(int i = 0; i < array.length; i++){
for(int j = 0; j < array[i].length; j++){
if(array[i][j] == target){
return true;
}
}
}
return false;
}
//二分查找(从左下角比较不断缩小范围)
public boolean findNumberIn2DArray2(int[][] array, int target){
int rows = array.length;
int cols = array[0].length;
int i = rows - 1;
int j = 0;
while(i >= 0 && j < cols){
if(array[i][j] == target){
return true;
}else if(array[i][j] > target){
i--;
}else{
j++;
}
}
return false;
}
}