问题
峰值元素是指其值大于左右相邻值的元素。
给你一个输入数组 nums,找到峰值元素并返回其索引。数组可能包含多个峰值,在这种情况下,返回任何一个峰值 所在位置即可。对于所有有效的 i 都有 nums[i] != nums[i + 1]
你可以假设 nums[-1] = nums[n] = -∞ 。
示例
输入: nums = [1,2,1,3,5,6,4]
输出: 1 或 5
解释: 你的函数可以返回索引 1,其峰值元素为 2;或者返回索引 5, 其峰值元素为 6。
解答
class Solution {
public:
int findPeakElement(vector<int>& nums) {
int left = 0, right = nums.size() - 1;
while (left < right) {
int mid = (left + right) / 2;
if (nums[mid] < nums[mid + 1]) left = mid + 1;
else right = mid;
}
return left;
}
};
重点思路
如果当前值比后一个值小,一定有峰值在当前值的右侧。以此开始二分。
需要注意的是,只有随便找一个峰值这种题才能二分。否则老老实实遍历吧。