问题描述
峰值元素是指其值大于左右相邻值的元素。给定一个输入数组 nums,其中 nums[i] ≠ nums[i+1],找到峰值元素并返回其索引。数组可能包含多个峰值,在这种情况下,返回任何一个峰值所在位置即可。你可以假设 nums[-1] = nums[n] = -∞
eg1:nums = [1,2,3,1],输出2,3是峰值元素,输出其索引2
eg2:nums = [1,2,1,3,5,6,4],输出1或者5,2和6都是峰值元素,索引分别为1和5
分析
a.遍历数组,如果当前元素大于它相邻的元素,则输出它的索引,若当前数组只有一个元素,则输出0,若当前元素为首尾元素,则只需要大于其另一侧的元素即可,时间复杂度为O(n)
b.二分法,每次取中间元素,判断其是否为峰值,是则输出mid,不是则判断峰值必然会存在与mid的左侧还是右侧,**一个常识是,当我们的目的地低于我们现在所处的高度,当我们上坡的时候,肯定是要下坡才能到达目的地,**那么将当前mid元素和mid-1元素相比,若mid-1元素值大于mid,则在mid-1—>0中一定存在峰值,否则峰值一定在mid+1–>right中,时间复杂度为O(logn)
int findPeakElement(vector<int>& nums) {
if(nums.size()==1) //只有一个元素则其必然是峰值
return 0;
if(nums[0]>nums[1]) //首尾为峰值的情况
return 0;
if(nums[nums.size()-1]>nums[nums.size()-2])
return nums.size()-1;
int left = 0;
int right = nums.size();
while(left<right)
{
int mid = (left+right)/2;
if(nums[mid]>nums[mid-1]&&nums[mid]>nums[mid+1])
return mid;
if(nums[mid]<nums[mid-1])
{
right = mid;
}
else
{
left = mid+1;
}
}
return left;
}