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

题目：

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

思路一：

针对二维矩阵的特点，可以注意到某个数的下方的数必定大于本身；而其左边的数必定小于本身；所以如果选取数组中右上角的数字为起点的话，作如下这些判断：如果该数字等于要查找的数字。查找过程即可结束；如果该数字大于要查找的数字，说明查找的数字在该数字的左边，也即必须剔除这个数字所在的列(列--)；当该数字小于要查找的数字，说明查找的数字在该数字的下边，也即剔除这个数字所在的行(行++)；这个过程可简单说：要查找的数字不在数组的右上角。每次查找都缩小查找范围，直到要查找的数字或者查找的范围为空。

public boolean findMatrix(int[][] matrix, int num) {
		boolean found = false;
		if (matrix == null || matrix.length == 0) {
			return found;
		}
		
		int row = 0;
		int col = matrix.length - 1;
		while (row < matrix.length && col >= 0) {
			if (matrix[row][col] > num) {
				--col;
			} else if (matrix[row][col] < num) {
				++row;
			} else {
				found = true;
				break;
			}
		}
		return found;
	}

思路二：二分查找

先对行进行二分查找，找到查找元素所在的行；需要注意的是当查找元素大于中间行元素时，要确定查找元素时在中间行，还是在下一行(中间行的最后列元素与查找元素相比，如果最后列元素小于查找元素，说明查找元素在下一行；如果列元素大于等于查找元素，说明查找元素在中间刚，查找结束)；

再对列进行二分查找，找到查找元素所在的列；在确定查找元素所在行时，剩下来只需要对这一行元素进行二分查找即可。

public static boolean findInMatrix(int[][] matrix, int num) {
		if (matrix == null || matrix.length == 0 || matrix[0].length == 0) {
			return false;
		}
		
		int r1 = 0, r2 = matrix.length - 1;
		while (r1 < r2) {
			int mid = r1 + (r2 - r1)/2;
			if (matrix[mid][0] > num) {
				r2 = mid - 1;
			} else if (matrix[mid][0] < num) {
				if (matrix[mid][matrix[0].length - 1] >= num) {
					r1 = mid;
					break;
				}
				r1 = mid + 1;
			} else {
				return true;
			}
		}
		System.out.println("the num row: " + r1);
		int c1 = 0, c2 = matrix[0].length - 1;
		while (c1 < c2) {
			int mid = c1 + (c2 - c1)/2;
			if (matrix[r1][mid] > num) {
				c2 = mid - 1;
			} else if (matrix[r1][mid] < num) {
				c1 = mid + 1;
			} else {
				return true;
			}
		}
		System.out.println("the num col: " + c1);
		return matrix[r1][c1] == num;
	}

扩展的题目：输出杨氏矩阵中的最小N的数；

改编题目：两个升序数组A和B，长度都是N。从两个数组中分别取出一个数，相加得到一个数，求N*N个和的前N个。

这道题目有两种做法：1）是通过堆来做，这对前N个元素要求排序比较好；2）是通过归并排序来做；具体的实现大家可以自己来写写看的。