《剑指offer》—— 4 二维数组中的查找

题目：

在一个二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。 请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。

思路

• 将二维数组看作矩阵，从右上角开始查找（row = 0，col = column.length - 1）；
• 如果比右上角的数字小，则排除这一列，col - 1；
• 如果比右上角的数字大，则排除这一行，row + 1；

代码

public class FindNumInMatrix {
 
	public static boolean find(int[][] arr, int number){
		
		if(arr == null){
			return false;
		}
		
		// 数组右上角的数字
		int column = arr[0].length - 1;   
		int row = 0;                      
		
		while(row < arr.length && column >= 0){
			if(arr[row][column] == number){
				return true;
			}else if(arr[row][column] > number){    //该数字小于右上角，排除该列
				column--;
			}else{                                  //数字大于右上角，排除该行
				row++;
			}
		}
		return false;
	}

	public static void main(String[] arge){
		int[][] arr = new int[4][4];
		array[0][0] = 1;
		array[0][1] = 2;
		array[0][2] = 3;
		array[0][3] = 4;
		array[1][0] = 5;
		array[1][1] = 6;
		array[1][2] = 7;
		array[1][3] = 8;
		array[2][0] = 9;
		array[2][1] = 10;
		array[2][2] = 11;
		array[2][3] = 12;
		array[3][0] = 13;
		array[3][1] = 14;
		array[3][2] = 15;
		array[3][3] = 16;
		System.out.println(find(array,6));
	}
}