寻找峰值
峰值元素是指其值严格大于左右相邻值的元素。
给你一个整数数组 nums
,找到峰值元素并返回其索引。数组可能包含多个峰值,在这种情况下,返回 任何一个峰值 所在位置即可。
你可以假设 nums[-1] = nums[n] = -∞
。
你必须实现时间复杂度为 O(log n)
的算法来解决此问题。
示例 1:
输入:nums = [
1,2,1,3,5,6,4]
输出:1 或 5
解释:你的函数可以返回索引 1,其峰值元素为 2;
或者返回索引 5, 其峰值元素为 6。
经典二分查找问题:
因为它的最左最右边是 负无穷,所以说一定存在峰值。
if nums[mid] < nums[mid + 1] //说明 mid 点后方一定存在峰值,否则为前方
接下来就是考虑 二分查找模板问题了;
如果是 if nums[mid] < nums[mid + 1] 这种情况,说明当前 mid 点肯定不是咱要找的点,l = mid + 1就很好理解;否则的话,就是说,峰值在 mid 前面,而 mid 点可能就是个峰值,所以说 r = mid 而不能 r = mid - 1;
代码:
class Solution { public: int findPeakElement(vector<int>& nums) { int l = 0, r = nums.size() - 1; while (l < r) { int mid = l + ((r - l) >> 1); if (nums[mid] < nums[mid+1]) { l = mid + 1; } else { r = mid; } } return r;//最后推出循环条件是 l == r 这里返回那个都无所谓 } };