162. 寻找峰值
峰值元素是指其值严格大于左右相邻值的元素。
给你一个整数数组 nums,找到峰值元素并返回其索引。数组可能包含多个峰值,在这种情况下,返回 任何一个峰值 所在位置即可。
你可以假设 nums[-1] = nums[n] = -∞ 。
你必须实现时间复杂度为 O(log n) 的算法来解决此问题。
示例 1:
输入:nums = [1,2,3,1]
输出:2
解释:3 是峰值元素,你的函数应该返回其索引 2。
这题很有意思,记录一下。
class Solution2 {
private:
int n = 0;
int res = 0;
public:
int findPeakElement(vector<int>& nums) {
n = nums.size();
int l = -1, r = n;
while(l+1 < r) {
int mid = (l+r) >> 1;
if (mid < n-1 && nums[mid] < nums[mid+1]) {
l = mid;
} else {
r = mid;
}
}
return r;
}
};
// 这种思路是思考什么时候mid+1会越界,其实只有mid=n-1, 那么,会导致mid=n-1的其实只有一种情况:l=n-2, r=n。所以这种情况直接让r=mid(其实就是r--)就好了。所以等价于:
class Solution {
private:
int n = 0;
int res = 0;
public:
int findPeakElement(vector<int>& nums) {
n = nums.size();
int l = -1, r = n;
while(l+1 < r) {
int mid = (l+r) >> 1;
if (mid == n-1) {
r = mid;
} else if (nums[mid] < nums[mid+1]) {
l = mid;
} else {
r = mid;
}
}
return r;
}
};

浙公网安备 33010602011771号