LeetCode74. 搜索二维矩阵

Posted 2021-06-10 mx_info

LeetCode74. 搜索二维矩阵

题目描述

/**
     * 编写一个高效的算法来判断 m x n 矩阵中，是否存在一个目标值。该矩阵具有如下特性：
     * <p>
     * 每行中的整数从左到右按升序排列。
     * 每行的第一个整数大于前一行的最后一个整数。
     */

思路分析

1. 由题目知，二维数组中的元素是有序的，因此可以使用二分查找法
2. 对二维数组使用二分查找，中间元素的索引计算方法不变，只是在取中间值时，应稍微发生变化，应该按照二维数组的行和列来取
3. 考虑如何判断一个索引在 m * n 的二维数组中的位置，则这个索引的行位置 应该为 索引 / n, 列位置为 索引 % n
4. 因此很容易判断该索引在二维数组中对应的值
5. 源码见下

源码及分析

/**
     * 
     * @param matrix 矩阵，及二维数组
     * @param target 要查找的目标值
     * @return 返回查找的结果
     */
    public boolean searchMatrix(int[][] matrix, int target) {
        //保存二维数组的行和列
        int row = matrix.length;
        int col = matrix[0].length;
        //对矩阵进行二分查找
        int left = 0, right = row * col - 1;
        while (left <= right) {
            //定义变量保存中间元素索引
            int mid = (right + left) / 2;
            //保存中间值
            int midVal = matrix[mid / col][mid % col];
            //二分判断
            if (midVal > target) {
                right = mid - 1;
            } else if (midVal < target) {
                left = mid + 1;
            } else {
                return true;
            }
        }
        return false;
    }