剑指 Offer 04. 二维数组中的查找（自我总结）

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

原题地址： https://leetcode-cn.com/problems/er-wei-shu-zu-zhong-de-cha-zhao-lcof/

在一个 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。

思路一：全部对比

最简单的方法，将输入元素同二维数组中的全部元素进行对比，只需要用2个for循环就可以完成

class Solution {
    public boolean findNumberIn2DArray(int[][] matrix, int target) {
        for(int i = 0 ; i < matrix.length ; i ++){
            for(int j = 0 ; j < matrix[0].length ; j ++){
                if(matrix[i][j] == target){
                    return true;
                }
            }
        }
        return false;
    }
}

但这种方法在，二维数据巨大的情况下，非常浪费运算。

方法二：二叉数算法

 

可以把整个二维数组看成一个二叉数，右上角的15 或者 左下角的18

看成二叉树的头部，分别连接着一个比自己大的数，和一个比自己小的数

class Solution {
    public boolean findNumberIn2DArray(int[][] matrix, int target) {
        
         
        int i = 0, j = matrix[0].length - 1;
        while (i < matrix.length && j >= 0) {
            int num = matrix[i][j];
            if (num == target) {
                return true;
            } else if (num > target) {
                j--;
            } else {
                i++;
            }
        }
        return false;
    }
}

从右上方开始比较

如果 num 和 target 相等，返回 true；如果 num 大于 target，列下标减 1；如果 num 小于 target，行下标加 1；循环体执行完毕仍未找到元素，返回false。