寻找峰值II

1. 题目

1901.寻找峰值II

一个 2D 网格中的 峰值 是指那些 严格大于 其相邻格子(上、下、左、右)的元素。

给你一个 从 0 开始编号 的 m x n 矩阵 mat ，其中任意两个相邻格子的值都 不相同 。找出 任意一个 峰值 mat[i][j] 并 返回其位置 [i,j] 。

你可以假设整个矩阵周边环绕着一圈值为 -1 的格子。

要求必须写出时间复杂度为 O(m log(n)) 或 O(n log(m)) 的算法

2. 示例

示例 1:

输入: mat = [[1,4],[3,2]]
输出: [0,1]
解释: 3 和 4 都是峰值，所以[1,0]和[0,1]都是可接受的答案。

示例 2:

输入: mat = [[10,20,15],[21,30,14],[7,16,32]]
输出: [1,1]
解释: 30 和 32 都是峰值，所以[1,1]和[2,2]都是可接受的答案。

提示：

• m == mat.length
• n == mat[i].length
• 1 <= m, n <= 500
• 1 <= mat[i][j] <= 105
• 任意两个相邻元素均不相等.

3. 思路

这题其实我当时是准备暴力的，但看到这题有一个提示要使用O(nlogn)算法后打消了这个念头。于是想到了二分查找。假如这个一共有m行n列，我们取三个值,left,right,和一个中间值mid,后面我每次mid行的最大值索引都记为max_index,如果我的mat[mid][max_index] > mat[mid+1][max_index]，我们就将right设置为mid,否则我将left设置为mid+1，这样我们就能找到目标值的行。

4. 代码

class Solution:
    def findPeakGrid(self, mat: List[List[int]]) -> List[int]:
        left , right = 0, len(mat) - 1    # 定义一个left和right
        # 二分查找
        while left < right:
            # 找到中间值并取整
            mid = (left + right) // 2
            # 当前mid行中最大值的列索引
            max_ = mat[mid].index(max(mat[mid]))
            # 如果当前行的最大值大于mid+1行的同列值，则后面找[left,mid]区间
            if mat[mid][max_] > mat[mid+1][max_]:
                right = mid 
            # 否则找[mid+1,right]区间
            else:
                left = mid + 1
        # 返回最终目的行的行索引和列索引
        return [left,mat[left].index(max(mat[left]))]