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

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

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

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

示例:

现有矩阵 matrix 如下：

[
  [1,   4,  7, 11, 15],
  [2,   5,  8, 12, 19],
  [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

---

思路1：暴力法

这个题一时没想到什么妙方法，感觉暴力法是可行的：
遍历二维数组中的每一个一维数组，
如果i < 目标值target，则i++；
如果i = target，则返回true;
如果i > target，则说明后面的都大于目标值，不需要比较，直接进入下一个数组

class Solution {
    public boolean findNumberIn2DArray(int[][] matrix, int target) {
        //判断边界值
        if(matrix == null || matrix.length == 0) return false;
        //去除[[]]这种数据
        if(matrix[0].length == 0) return false;
        
		//matrix.length 行数
        for(int i = 0; i < matrix.length; i++){//i代表第i行
            for(int j = 0; j < matrix[i].length; j++){//j代表第j列
                if(matrix[i][j] == target){//matrix[i][j]，第i行第j列
                    return true;
                }else if(matrix[i][j] > target){
                    break;//去下一行去找
                }
            }
        }
        return false;
    }
}

时间复杂度：O(n*m)
空间复杂度：O(1)

上述代码没有利用到列也是递增的这个信息，因此，代码还可以优化下。
充分利用列递增的信息：如果matrix[i][0] > target，则后面不需要再做比较，直接返回false

class Solution {
    public boolean findNumberIn2DArray(int[][] matrix, int target) {
        //判断边界值
        if(matrix == null || matrix.length == 0) return false;
		if(matrix[0].length == 0) return false;
		
        for(int i = 0; i < matrix.length; i++){
        	if(matrix[i][0] > target){//第i行头一个，如果大于目标，后面不需要找了
                    return false;
            }
            for(int j = 0; j < matrix[i].length; j++){
                if(matrix[i][j] == target){
                    return true;
                }else if(matrix[i][j] > target){
                    break;
                }
            }
        }
        return false;
    }
}

---

思路2：右上角开始

书上的思路是：从右上角开始
首先，选取数组中右上角的数字，如果该数字等于要查找的数字，则查找过程结束；
如果该数字大于要查找的数字，则剔除这个数字所在的列；
如果该数字小于要查找的数字，则剔除这个数字所在的行；
这样，每一步都在数组的查找范围中剔除一行或一列，每一步都可以缩小查找范围，直到找到要查找的数字，或者查找范围为空。

class Solution {
    public boolean findNumberIn2DArray(int[][] matrix, int target) {
        //判断边界值
        if(matrix == null || matrix.length == 0) return false;
        if(matrix[0].length == 0) return false;

        int col = matrix[0].length;
        //右上角：i从0开始，j从matrix[0].length - 1开始
        for(int i = 0; i < matrix.length; i++){
            for(int j = col - 1; j >= 0; j--){
                if(matrix[i][j] == target){
                    return true;
                }else if(matrix[i][j] > target){//去掉该列
                    col--;
                }else if(matrix[i][j] < target){//去掉该行
                    break;
                }
            }
        }

        return false;
    }
}

时间复杂度：
由于每次都能去掉一行或一列，最坏时间复杂度为：O(n + m)
空间复杂度：O(1)