题目:在一个二维数组中,每一行都按照从左到右递增的顺序排序,、每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
- 解决问题:首先遍历是可行的,左上角出发,先行后列,从左到右,从上到下。(效率O(m*n))
- 多种解法:书上的解法是从右上角出发,每一次比较都可以舍弃一行或一列,类似二分法。(效率O(m*n))
- 方法分析:首先矩阵的数字排布是有规律的,如何充分的利用题中的规律,是高效率的关键。观察矩阵,斜对角线的元素正好为途中L型数列的中数,从右上角出发每次比较都可以排除一些元素,但从左上角出发则不可以。
如下图矩阵中查找数字5
public class FindMaxInMatrix {
/* 二维数组中的查找:在一个二维数组中,
每一行都按照从左到右递增的顺序排序,
每一列都按照从上到下递增的顺序排序。
请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。*/
public static void main(String[] args) {
int[][] matrix = {{1, 2, 8, 9}, {2, 4, 9, 12 },
{4, 7, 10, 13}, {6, 8, 11, 15}};
int target = 150;
System.out.println(FindMax1(matrix, target));
}
// 遍历二维数组,时间复杂度O(n*m)
// 可以优化为当从左到右遍历行的时候若超过,则直接到下一列
// 也就是选取左上角的数字,从左到右,从上到下进行遍历
public static boolean FindMax1(int[][] matrix, int target) {
if(matrix.length == 0) {
return false;
}
int num;
for(int i = 0; i < matrix.length; i++) {
for(int j = 0; j < matrix[i].length; j++) {
num = matrix[i][j];
if(num < target) {
continue;
}else if(num > target) {
break;
}else {
return true;
}
}
}
return false;
}
// 剑指Offer方法:
// 选取数组中右上角的数字。从右到左,从上到下进行遍历、
// 则比其第一种方法,每一次都在数组的查找范围中剔除一行或者一列,
// 这样每一步都可以缩小查找的范围,直到找到要查找的数字。
// 因为观察矩阵的副对角线,把数字划分为了大小两部分,类似于二分法。
// 而主对角线则没有这种优势
public static boolean FindMax2(int[][] matrix, int target) {
if(matrix.length == 0) {
return false;
}
int num;
for(int i = 0; i < matrix.length; i++) {
for(int j = matrix[i].length-1; j >= 0; j--) {
num = matrix[i][j];
if(num < target) {
break;
}else if(num > target) {
continue;
}else {
return true;
}
}
}
return false;
}
}