博客主页: A_SHOWY
系列专栏:力扣刷题总结录 数据结构 云计算 数字图像处理 力扣每日一题_
1901. 寻找峰值 IIhttps://leetcode.cn/problems/find-a-peak-element-ii/
看到这个时间复杂度就知道和昨天的题目一样要用二分了,比昨天的难度稍微大一点,这个题目是一个二维找峰值了。不过整体思路差不多。
我们要明确如果i1行的最大值比它上面的格子大,同时i2行比它下面的格子小,i1<=i2.那么在【i1,i2】之间一定存在峰值。我们可以用二分法求解
构造二分法:设置low为0,high为m-1,m为这个数组的行数。并且设置i(mid)为二分之low+high。第i行的最大值为第j列。
如果mat[i][j] < mat[i-1][j],那么high = i-1; 继续‘
如果mat[i][j] < mat[i+1][j],那么low = i + 1; 继续
返回(i,j ) 的结果
class Solution {
public:
vector findPeakGrid(vector>& mat) {
int m = mat.size();
int low = 0;int high = m -1;
while(low <= high){
int i = (low + high)/2;
int j = max_element(mat[i].begin(),mat[i].end()) - mat[i].begin();//找到第i行的最大值
if(i - 1 >=0 &&mat[i][j] < mat[i-1][j]) {
high = i - 1;
continue;
}
if(i + 1 < m && mat[i][j] < mat[i+1][j]){
low = i + 1;
continue;
}
return {i,j};
}
return{};
}
};
int j = max_element(mat[i].begin(),mat[i].end()) - mat[i].begin();//找到第i行的
if(i - 1 >=0 &&mat[i][j] < mat[i-1][j]) {
high = i - 1;
continue;
}
if(i + 1 < m && mat[i][j] < mat[i+1][j]){
low = i + 1;
continue;
}
注意在第一个if中由于high是i-1,因为是减法操作所以要保证i-1大于等于下边界0
而第二个if中low是i+1,不能超过它的上边界m,这才是if中需要注意的条件