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

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

class Solution {
        /**
         * 优化问题 利用二分查找,先去判断在那一行,然后再利用二分查找返回判断是否在此行数据中
         * 第一次行数的判断只能缩小查找的范围,并不能完成确定在哪一行。
         * 最朴素的方法最行之有效,应该使用线性查找
         *
         * @param matrix 二维数组
         * @param target 目标数字
         * @return 是否存在
         */
        public boolean findNumberIn2DArray(int[][] matrix, int target) {
            if (matrix == null || matrix.length == 0 || matrix[0].length == 0) {
                return false;
            }
            int row = 0, list = 0, column = matrix[0].length - 1, rowLen = matrix.length, listLen = matrix[0].length;
            while (row < rowLen && column >= 0) {
                int num = matrix[row][column];
                if (num == target) {
                    return true;
                } else if (num > target) {
                    column--;
                } else if (num < target) {
                    row++;
                }
            }
            return false;
        }
    }
经验分享 程序员 微信小程序 职场和发展