描述
给定一个长度为n的数组nums,请你找到峰值并返回其索引。数组可能包含多个峰值,在这种情况下,返回任何一个所在位置即可。
1.峰值元素是指其值严格大于左右相邻值的元素。严格大于即不能有等于
2.假设 nums[-1] = nums[n] = -\infty−∞
3.对于所有有效的 i 都有 nums[i] != nums[i + 1]
4.你可以使用O(logN)的时间复杂度实现此问题吗?
数据范围:
1 \le nums.length \le 2\times 10^5 \1≤nums.length≤2×105
-2^{31}<= nums[i] <= 2^{31} - 1−231<=nums[i]<=231−1
如输入[2,4,1,2,7,8,4]时,会形成两个山峰,一个是索引为1,峰值为4的山峰,另一个是索引为5,峰值为8的山峰,如下图所示:
思路:时间复杂度要求O(logN),由此联想到二分查找法,这里能二分查找,是因为:对于所有有效的 i 都有 nums[i] != nums[i + 1]&&最边上都是无穷小。找到mid后就出现了3种情况,直接就找到了山峰,除此以外就向高的地方爬,一定可以找到一个峰值。(如果一直增加的话,峰值就出现在最后一个数)
class Solution {
public:
int findPeakElement(vector<int>& nums) {
// write code here
if(nums.size()==1||nums[0]>nums[1]) return 0;
int left=0,right=nums.size()-1,mid;
if(nums[right-1]<nums[right]) return right;
while(right>left)
{
mid=left+(right-left)/2;
if(nums[mid]>nums[mid-1]&&nums[mid]>nums[mid+1]) return mid;
if(nums[mid]>nums[mid+1]) right=mid;
if(nums[mid]<nums[mid+1]) left=mid;
}
return -1;
}
};