题目描述
峰值元素是指其值大于左右相邻值的元素。
给定一个输入数组 nums,其中
n
u
m
s
[
i
]
≠
n
u
m
s
[
i
+
1
]
nums[i] \neq nums[i+1]
nums[i]̸=nums[i+1],找到峰值元素并返回其索引,假设
n
u
m
s
[
−
1
]
=
n
u
m
s
[
n
]
=
−
∞
nums[-1] = nums[n] = -\infty
nums[−1]=nums[n]=−∞。
数组可能包含多个峰值,在这种情况下,返回任何一个峰值所在位置即可。
解题思路
本题目属于比较难的二分算法的应用,这种题目虽然没有二段性质,但是可以使用二分思想进行解题。
本题目假定
n
u
m
s
[
−
1
]
=
n
u
m
s
[
n
]
=
−
∞
nums[-1] = nums[n] = -\infty
nums[−1]=nums[n]=−∞,那么容易得知,一定存在峰值元素 (因为两边都是负无穷,其中定有元素值大于左右相邻元素)。
根据此理,可以得到如下推论:假定
n
u
m
s
[
m
i
d
]
>
n
u
m
s
[
m
i
d
+
1
]
nums[mid] > nums[mid + 1]
nums[mid]>nums[mid+1],那么
[
l
,
m
i
d
]
[l,mid]
[l,mid] 区间范围内一定存在一个峰值元素。
实现代码
int findPeakElement(vector<int>& nums)
{
int l = 0,r = nums.size() - 1;
while(l < r)
{
int mid = (l + (long long)r) >> 1;
if(nums[mid] > nums[mid + 1])
{
r = mid;
}
else
{
l = mid + 1;
}
}
return l;
}