剑指 offer 刷题——04. 二维数组中的查找

一、题目

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

示例:

现有矩阵 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

二、思路及代码实现

题目中给定的数组有两个特点：

• 每一行都按照从左到右递增的顺序排序；
• 每一列都按照从上到下递增的顺序排序。

那显然不是让我们从头开始逐个比较来进行查找的。

把二维数组看成矩阵，则左下角的这个数，是这一行最小的数，是这一列最大的数。我们从左下角开始查找，通过比较左下角数字与目标值的大小来决定是向上还是向右查找，这样的话，每次可排除一整行或一整列。

以示例为例进行分析，我们要查找数字 5，从左下角开始。

• 左下角数字 18 大于目标值 5，则18 所在的这一行都比 5 大，所以排除这一行，去 18 的上面一行查找；
• 此时的左下角数字是 10，仍然大于 5，再去 10 的上面一行查找；
• 此时的左下角数字是 3，小于目标值 5，所以排除 3 这一列，去 3 的右边一列查找；
• 此时左下角数字 6 大于目标值 5，再去 6 的上面一行查找，左下角数字为 5，成功找到。

同样地，我们也可以从矩阵的右上角开始查找。

参考代码：

class Solution {
    public boolean findNumberIn2DArray(int[][] matrix, int target) {
        if(matrix.length == 0 || matrix[0].length == 0)
            return false;
        int row = matrix.length - 1;
        int col = 0;
        // 从左下角开始查找
        while(row >= 0 && col < matrix[0].length){
            if(target == matrix[row][col])
                return true;
            else if(target > matrix[row][col])
                col++;
            else
                row--;
        }
        return false;
    }
}