find-peak-element-ii

一个整数矩阵有如下一些特性:

  • 相邻的整数都是不同的
  • 矩阵有 n 行 m 列。
  • 对于所有的 i < m, 都有 A[0][i] < A[1][i] && A[n - 2][i] > A[n - 1][i].
  • 对于所有的 j < n, 都有 A[j][0] < A[j][1] && A[j][m - 2] > A[j][m - 1].

我们定义一个位置 P 是一个峰,如果有 A[j][i] > A[j+1][i] && A[j][i] > A[j-1][i] && A[j][i] > A[j][i+1] && A[j][i] > A[j][i-1]。

找出该矩阵的一个峰值元素,返回他的坐标。

您在真实的面试中是否遇到过这个题?  
Yes
样例

给一个矩阵:

[
  [1 ,2 ,3 ,4 ,5],
  [16,41,23,22,6],
  [15,17,24,21,7],
  [14,18,19,20,8],
  [13,12,11,10,9]
]

返回 41 的坐标[1,1], 或者 24 的坐标[2,2]

注意

可能会存在多个峰值,返回任意一个即可

class Solution {
public:
    vector<int> findPeakII(vector<vector<int> > A) {
        int n=A.size();
        int m=A[0].size();
        int low=0,high=n-1;
        int rowmid,col;
        vector<int>res;
        while(low<=high){
            rowmid=low+(high-low)/2;
            col=findPeak2(A[rowmid]);
            if(A[rowmid][col]<A[rowmid+1][col]) low=rowmid+1;
            else if(A[rowmid][col]<A[rowmid-1][col]) high=rowmid-1;
            else return {rowmid,col};
        }
    }
    /*
    int findPeak(vector<int>A){
        int n=A.size();
        int left=0;
        int right=n-1;
        while(left<right){
            int mid=left+(right-left)/2;
            if(A[mid]<A[mid+1]) left=mid+1;
            else right=mid;
        }
        return right;
    }
    */
    int findPeak2(vector<int> &v) {
        int res = 0;
        for (int i = 1; i < v.size(); ++i) {
            if (v[i] > v[res]) res = i;
        }
        return res;
    }
};


你可能感兴趣的:(find-peak-element-ii)