剑指offer 面试题3 二维数组中的查找

剑指offer 面试题3 二维数组中的查找

/**
* Created by liyazhou on 2017/5/22.
*/
public class Test3 {

public static boolean contains(int[][] matrix, int number){
for (int i = 0; i < matrix.length; i++)
for (int j = 0; j < matrix[i].length; j++)
if (matrix[i][j] == number) {
System.out.println(String.format("i, j = %d, %d", i, j));
return true;
}
return false;
}

public static boolean contains2(int[][] matrix, int number){
boolean contains = false;
int col = matrix[0].length-1;
int row = 0; // 右上角元素的
if (number < matrix[0][0] || number > matrix[matrix[0].length-1][matrix.length-1]) return false;  // 判断边界
while(row < matrix.length && col >= 0){
int topRightVal = matrix[row][col];
if (number == topRightVal) {
contains = true; break;
}
else if (number < topRightVal) {  // 删除当前元素所在列
--col;
}else{ // number > topRightVal，删除当前元素所在行
++row;
}
}
System.out.println(String.format("row, col = %d, %d", row, col));
return contains;
}

/**
* 在一个二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。
* 请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。
* <p/>
* 规律：
*  首先选取数组中右上角的数字。如果该数字等于要查找的数字，查找过程结束。
*  如果该数字大于要查找的数字，剔除这个数字所在的列；
*  如果该数字小于要查找的数字，剔除这个数字所在的行。
*  也就是说如果要查找的数字不在数组的右上角，则每一次都在数组的查找范围中剔除行或者列，
*  这样每一步都可以缩小查找范围，直到要查找的数字，或者查找范围为空。
* @param matrix 二维数组
* @param number 待查找的数字
* @return 在数组中是否找到该数字
*/
public static boolean contains3(int[][] matrix, int number){
if (matrix == null || matrix.length < 1 || matrix[0].length < 1) return false;  // 判断数组
int rows = matrix.length;
int cols = matrix[0].length;
if (number < matrix[0][0] || number > matrix[rows-1][cols-1]) return false;  // 判断边界

boolean found = false;
int row = 0;
int col = cols-1;  // todo, 注意边界
while(row < rows && col >=0){
int topRightVal = matrix[row][col];  // 获取到查找范围内的右上角的元素
if (number == topRightVal){
found = true; break;  // 找到该数字，停止查询
}
else if (number < topRightVal) --col;  // 删除当前列， 第 col 列
else ++row;  // number > topRightVal， 删除当前行，第 row 行
}
System.out.println(String.format("row, col = %d, %d", row, col));
return found;
}

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[] numbers = {-1, 1, 15, 7};
for (int number : numbers){
System.out.println(String.format("contains  %d, %b", number, Test3.contains(matrix, number)));
System.out.println(String.format("contains2 %d, %b", number, Test3.contains2(matrix, number)));
System.out.println(String.format("contains3 %d, %b", number, Test3.contains3(matrix, number)));
}
}
}