9.7 1901. 寻找峰值 II(中等)

1901. 寻找峰值 II - 力扣（LeetCode）

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

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

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

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

示例 1:

输入: mat = [[1,4],[3,2]]

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

思路：因为有时间效率问题，暴力遍历会超出限制，可先选择一行，找出改行的最大值，再对比其上元素，若大于上元素则自身或者其下半区一定存在峰值。则可使用二分法，以行为单位进行淘汰。

class Solution {
public:
    vector<int> findPeakGrid(vector<vector<int>>& mat) {
        int n = mat.size();
        int m = mat[0].size();
        int low=0,high=n-1;
        while(low<high)
        {
            int mid=(low+high)/2;
            //找到mid行中的最大值下标
            int maxIndex=max_element(mat[mid].begin(),mat[mid].end())-mat[mid].begin();
            //判断该值是否大于该值的下一行
            if(mat[mid][maxIndex]<mat[mid+1][maxIndex])
            {
                //那mid的下半区一定会存在峰值
                low=mid+1;
            }
            else{
                //mid自身行或者上半区有峰值
                high=mid;
            }
        }
        //找出此行的最大值
        int maxIndex=max_element(mat[low].begin(),mat[low].end())-mat[low].begin();
        return {low,maxIndex};
    }
};

方法二：

思路：此处峰值定义为大于自身的上下左右元素，若此数不是峰值可以选择上下左右中最大元素再进行判断，依次找到峰值

class Solution {
public:
    vector<int> findPeakGrid(vector<vector<int>>& mat) {
        int n=mat.size(),m=mat[0].size();
        int i=n/2,j=m/2;
        //假设从中间位置开始找
        while(i<n&&i>=0&&j>=0&&j<m)
        {
            int up=i-1<0?-1:mat[i-1][j];
            int down=i+1>=n?-1:mat[i+1][j];
            int left=j-1<0?-1:mat[i][j-1];
            int right=j+1>=m?-1:mat[i][j+1];
            //找到那个更大的值
            if(mat[i][j]>up&&mat[i][j]>down&&mat[i][j]>left&&mat[i][j]>right) return {i,j};
            //此时一定要进行更新i，j（i，j并不会超出矩阵，因为-1不会是最大值）
            int temp=max({up,down,left,right});
            if(up==temp) i--;
            else if(right==temp) j++;
            else if(down==temp) i++;
            else j--;
        }
        //并不会走到这
        return {};
    }
};