二维数组中的查找

最新推荐文章于 2024-08-22 21:24:01 发布

天将降大任于我

最新推荐文章于 2024-08-22 21:24:01 发布

阅读量110

点赞数

分类专栏：算法文章标签：算法 leetcode java

本文链接：https://blog.csdn.net/smznbhh/article/details/126969056

版权

算法专栏收录该内容

16 篇文章 0 订阅

订阅专栏

问题描述：

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

限制：0 <= n <= 1000 0 <= m <= 1000

例子

示例 1：
现有矩阵 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。

提示：

无

代码实现：

1.循环二分：

每一行用二分查找

时间复杂度：O（MlogN）空间复杂度：O（1）

public static boolean findNumberIn2DArray(int[][] matrix, int target) {
            if (matrix.length == 0 || matrix[0].length == 0) return false;
            for (int[] ints : matrix) {// 行
                int left = 0;
                int right = matrix[0].length - 1;
                while (left <= right) {// 二分
                    int half = left + ((right - left) >> 1);
                    if (ints[half] < target) {
                        left = half + 1;
                    } else if (ints[half] > target) {
                        right = half - 1;
                    } else return true;
                }
            }
            return false;
        }

2.仿搜索树：

从左下角或者右上角开始找，因为每行开头和结尾的值是极端，相当于搜索二叉树的节点；从左下角找，比目标数大就往上一行找，比目标数小就往下一列找

时间复杂度：O（M+N）空间复杂度：O（1）

public static boolean findNumberIn2DArray2(int[][] matrix, int target) {
            if (matrix.length == 0 || matrix[0].length == 0) return false;
            int i = 0;
            int j = matrix.length - 1;
            while (j >= 0) { // 从左下角开始找（也可以从右上角，i--,j++）
                if (matrix[j][i] > target) j--;// 开头的值大于目标数，往上一行找
                else if (matrix[j][i] < target) {// 开头的数小于目标数，往下一列找
                    ++i;
                    if (i > matrix[0].length - 1) return false;
                } else return true;
            }
            return false;
        }