剑指 Offer 04. 二维数组中的查找——以列为标志查找目标数

剑指 Offer 04. 二维数组中的查找

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

示例:

现有矩阵 matrix 如下：

[
	[1,   4,  7, 11, 15],
	[3,   6,  9, 16, 22],
	[10, 13, 14, 17, 24],
	[18, 21, 23, 26, 30]
]

给定 target = 5，返回 true。

给定 target = 20，返回 false。

限制：

0 <= n <= 1000

0 <= m <= 1000

class Solution {
    public boolean findNumberIn2DArray(int[][] matrix, int target) {
    	//边界值判断，防止matrix数组的行或列为空
        if (matrix == null || matrix.length == 0 || matrix[0].length == 0) {
            return false;
        }
        for(int i = 0; i < matrix[0].length; i++){
        	//循环判断每一列的开头元素和结尾元素，找到目标元素的目标列
            if(target >= matrix[0][i] && target <= matrix[matrix.length - 1][i]){
                for(int j = 0; j < matrix.length; j++){
                    if(matrix[j][i] == target){
                        return true;
                    }
                }
            }
        }
        return false;
    }
}

首先循环判断matrix每一列的第一个元素和第二个元素，因为二维数组是每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序的，所以目标元素一定大于列首一定小于列尾。找到所处列之后再在列中寻找目标元素。
时间复杂度为O（n^2)