题目:在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
这题比较简单,以图中二维数组为例。假设我们输入的整数为num,要想判断数组中是否含有num,那么肯定需要从二维数组某个数字开始和num一一比对。哪个数字开始呢?经分析应该从左上角或右下角的数字开始比较判断。
举例分析:(假设输入数字num为7)
从右上角数字9开始,7比9小,7不可能在数字9这一列了,我们就剔除掉数字9所在列;
现在右上角的数字是8,7比8小,那再把数字8这一列给剔除掉;
现在右上角的数字是2,7比2大,2不可能在数字2这一行了,同样就剔除数字2所在行;
现在右上角的数字是4,7比4大,那再把数字4这一行给剔除掉;
现在右上角的数字是7,和num相等,表示数组中存在数字7。
简单而言就是,每次使用右上角的数字和输入数字num比较,右上角数字比num大就剔除当前列,比num小就剔除当前行。
代码如下:
public static void findIn2DArray(int[][] arr, int num) {
if (ObjectUtils.isEmpty(arr)) return;
int rows = arr.length;
int cols = arr[0].length;
int row = 0; // 右上角数字行下标
int col = cols - 1; // 右上角数字列下标
while (rows > row && col >= 0) {
if (arr[row][col] > num) {
col--;
} else if (arr[row][col] < num) {
row++;
} else {
System.out.println("yes");
return;
}
}
System.out.println("no");
}