二维数组中的查找

一、需求

1. 在一个 n * m 的二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。

2. 请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。

二、遍历法 

2.1  思路分析

1. 这个很简单，就像是查找二维数组中的元素一样；
2. 需要注意的是，要首先排除空数组的存在。

2.2  代码实现

public boolean findNumberIn2DArray(int[][] matrix, int target) {
        
        if(matrix.length == 0) {
            return false;
        }
        
        int rows = matrix.length;
        int cols = matrix[0].length;

        for(int i = 0; i < rows; i++) {
            for(int j = 0; j < cols; j++) {
                if(target == matrix[i][j]) {
                    return true;
                }
            }
        }
        return false;
    }

 2.3  复杂度分析

• 最坏情况下遍历所有数据，时间复杂度为O(mn)；
• 没有申请空间，空间复杂度为O(1)；

三、遍历法(优化)

3.1  思路分析

1. 从右上角第一个元素开始遍历，假设要找16，那么从15开始；
2. 发现15小于16，那么行坐标+1，从而移动到19；
3. 发现19大于16，那么列坐标-1，从而移动到16；

3.2  代码实现 

public boolean findNumberIn2DArray(int[][] matrix, int target) {
        //判断边界条件
        if(matrix == null || matrix.length == 0 || matrix[0] == null || 
        matrix[0].length == 0) {
            return false;
        }
        //获取数组的行、列值
        int rows = matrix.length;
        int cols = matrix[0].length;
        //定位到右上角的元素
        int i = 0;
        int j = cols - 1;
        //开始查找
        while(i < rows && j >= 0) {
            //当前值小于目标值，i+1
            if(matrix[i][j] < target) {
                i++;
            } else if(matrix[i][j] > target) {
                j--;
            } else {
                return true;
            }
        }
        return false;
    }

3.3  复杂度分析

• 时间复杂度为O(n)；
• 空间复杂度为O(1)；