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

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

 

思路：

每一次都是选取数组查找范围的右上角数字。如果该数字等于要查找的数字，查找过程结束；如果该数字大于要查找的数字，剔除这个数字所在的列；如果该数字小于要查找的数字，剔除这个数字所在的行。
也就是说如果要查找的数字不在数组的右上角，则每一次都在数组的查找范围中剔除一行或者一列。这样可以每一步都可以缩小查找的范围，直到找到要查找的数字，或者查找范围为空。

public boolean findInPartiallySortedMatrix(int[][] matrix, int target) {
	if (matrix == null || matrix.length == 0)
		return false;
	int row = 0, col = matrix[0].length;
	while (row < matrix.length || col > 0) {
		if (matrix[row][col] == target)
			return true;
		else if (matrix[row][col] < target) {
			row++;
		} else if (matrix[row][col] > target) {
			col--;
		}
	}
	return false;
}

 

def findNumberIn2DArray(matrix, target):
    """
    :type matrix: List[List[int]]
    :type target: int
    :rtype: bool
    """
    if matrix == [] or matrix == [[]]:
        return False
    i = 0
    j = len(matrix[0])-1
    while i<len(matrix) and j>=0:
        if matrix[i][j] < target:
            i += 1
        elif matrix[i][j] > target:
            j -= 1
        elif matrix[i][j] == target:
            return True
    return False