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

思考

• 一看到 我就十分暴力的想从左上角开始查找
• 但是写出来的就是一个巨大的bug
• 有很多没有考虑到的情况
• 然后想二分，但是又不会写
• 看到题解有人说可以看成一个有序二叉树！！！突然get到直接从右上角查找即可
  • 如果target小于当前节点，则查左子树，即col--；
  • 如果target大于当前节点，则查找右子树，即row++
  • 如果target等于当前节点，则return true
  • 边界判断，如果越界则return false
• 关于vector
  • 一开始用c++写的时候，对vector的操作很陌生，不知道怎么简单操作下标的移动，一开始就只知道用iterator，然后发现可以用size()
• java实现
  • 然后用Java实现的时候，发现我基本上对Java一无所知
    • 输入是二维数组，下标可以直接用符号。
    • 长度的话，可以用length成员，不用括号！！
    • 不能用‘!’来看是不是0？

代码

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