一个整数矩阵有如下一些特性:
- 相邻的整数都是不同的
- 矩阵有 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;
}
};