Leetcode 1901. 寻找峰值 II(Java + 列最大值 + 二分)

2023-12-20 06:32:40

题目

  • 1901. 寻找峰值 II
  • 一个 2D 网格中的 峰值 是指那些 严格大于 其相邻格子(上、下、左、右)的元
  • 给你一个 从 0 开始编号 的 m x n 矩阵 mat ,其中任意两个相邻格子的值都 不相同 。找出 任意一个 峰值 mat[i][j] 并 返回其位置 [i,j] 。
  • 你可以假设整个矩阵周边环绕着一圈值为 -1 的格子。
  • 要求必须写出时间复杂度为 O(m log(n)) 或 O(n log(m)) 的算法
  • m == mat.length
  • n == mat[i].length
  • 1 <= m, n <= 500
  • 1 <= mat[i][j] <= 10 ^ 5
  • 任意两个相邻元素均不相等.

解法

Java + 列最大值 + 二分

第 1 步:

  • 类似:Leetcode 162. 寻找峰值(Java + 二分)
  • 在行内找严格大于左右的元素,再找每列的最大值(一定是大于上下)
  • 一定需要找该列的最大值,如果这也二分找极大值(仅严格大于左右),那么可能找到非该列最大值从而导致 左/右 列误判

第 2 步:

  • 具体做法:
    • 先找中间=mid 列,找到俩最大值 mat[maxRow][mid] ,元素一定严格大于上下的元素
    • 如果 mat[maxRow][mid] 严格大于左右的元素,则直接返回,否则下一步
    • 如果 mat[maxRow][mid] > mat[maxRow][mid+1] 则 maxRow 左边列一定存在,否则 maxRow 右边列一定存在
  • 时间复杂度:O(m*logn),空间复杂度:O(1)

代码

/**
     * Java + 列最大值 + 二分:
     *
     * 第 1 步:
     * 类似:162. 寻找峰值 FindPeakElement,在行内找严格大于左右的元素,再找每列的最大值(一定是大于上下)
     * 一定需要找该列的最大值,如果这也二分找极大值(仅严格大于左右),那么可能找到非该列最大值从而导致 左/右 列误判
     *
     * 第 2 步:
     * 具体做法:
     *     * 先找中间=mid 列,找到俩最大值 mat[maxRow][mid] ,元素一定严格大于上下的元素
     *     * 如果 mat[maxRow][mid] 严格大于左右的元素,则直接返回,否则下一步
     *     * 如果 mat[maxRow][mid] > mat[maxRow][mid+1] 则 maxRow 左边列一定存在,否则 maxRow 右边列一定存在
     * 时间复杂度:O(m*logn),空间复杂度:O(1)
     *
     *
     */
    public int[] findPeakGrid(int[][] mat) {
        int leftCol = 0;
        int rightCol = mat[0].length - 1;
        int resCol = 0;
        while (leftCol <= rightCol) {
            int midCol = ((rightCol - leftCol) >> 1) + leftCol;

            int maxRow = getMaxRow(mat, midCol);
            if ((midCol == 0 || mat[maxRow][midCol] > mat[maxRow][midCol - 1])
                    && (midCol == mat[0].length - 1 || mat[maxRow][midCol] > mat[maxRow][midCol + 1])) {
                resCol = midCol;
                break;
            }

            if (midCol == mat[0].length - 1 || mat[maxRow][midCol] > mat[maxRow][midCol + 1]) {
                rightCol = midCol - 1;

            } else {
                leftCol = midCol + 1;
            }

        }

        return new int[]{getMaxRow(mat, resCol), resCol};
    }

    private int getMaxRow(int[][] mat, int resCol) {
        int maxRow = 0;

        for (int i = 0; i < mat.length; i++) {
            if (mat[maxRow][resCol] < mat[i][resCol]) {
                maxRow = i;
            }
        }

        return maxRow;
    }

文章来源:https://blog.csdn.net/qq_33530115/article/details/135097027
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。