剑指 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。

链接：https://leetcode-cn.com/problems/er-wei-shu-zu-zhong-de-cha-zhao-lcof

解题思路

• 把矩阵的左下角的节点作为根节点，开始搜索
• 循环终止条件：数组越界
• 如果当前值小于目标值，则向右搜索
• 如果当前值大于目标值，则向上搜索

算法流程

• row = matrix.length, clo = 0，从row和clo开始搜索
• 循环条件：row >= 0 并且 clo <matrix[0].length
  • matrix[row][clo] < target, clo++
  • matrix[row][clo] > target, row--
  • matrix[row][clo] > target, 返回true
• 循环结束，没有找到，返回false

代码

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