题目描述
难度:中等
峰值元素是指其值严格大于左右相邻值的元素。
给你一个整数数组 nums,找到峰值元素并返回其索引。数组可能包含多个峰值,在这种情况下,返回 任何一个峰值 所在位置即可。
你可以假设 nums[-1] = nums[n] = -∞ 。
你必须实现时间复杂度为 O(log n) 的算法来解决此问题。
示例 1:
输入:nums = [1,2,3,1]
输出:2
解释:3 是峰值元素,你的函数应该返回其索引 2。
示例 2:
输入:nums = [1,2,1,3,5,6,4]
输出:1 或 5
解释:你的函数可以返回索引 1,其峰值元素为 2;
或者返回索引 5, 其峰值元素为 6。
提示:
1 <= nums.length <= 1000
-231 <= nums[i] <= 231 - 1
对于所有有效的 i 都有 nums[i] != nums[i + 1]
分析
读完题目之后其实可以这样考虑,所谓峰值不就可以理解为数组中的最大值吗?我们可以直接通过遍历数组获取最大值的索引然后返回就行,但是这个算法的时间复杂度是 O( n ),如果用此算法提交的话是没有问题的,那么有没有时间复杂度是 O ( logn ) 的呢?
我在解题时翻阅了别人的题解,发现一种时间复杂度是 O ( logn ) 解法,就是利用二分查找的思维来解题;
题目给出条件 nums[-1] = nums[n] = -∞,基于这个条件,我们可以得出数组的两端都是无穷小,在这一点上进行思考的话,其实可以得出这样的结论,给定一个中点 mid ,若 mid + 1 位置的元素大于 mid 所在位置的元素,那么一定有峰值存在于 mid 的右边;反过来也是一样,若 mid + 1 位置的元素小于 mid 所在位置的元素,那么一定有峰值存在于 mid + 1 的左边;稍微推衍一下就可以得出这样的结论了。
有这样的结论之后,就可以开始编写代码了,利用二分查找的思路,不停的缩小峰值存在的数组范围,最后 left 和 right 重叠的位置就是峰值的位置。
代码:
public int findPeakElement(int[] nums) {
int left = 0;
int right = nums.length - 1;
while(left < right){
int mid = left + (right - left)/2;
if(nums[mid] > nums[mid + 1]){
right = mid;
}else {
left = mid + 1;
}
}
return left;
}
总结
最开始是完全没有头绪的,后来了解到可以直接返回最大值作为峰值,后翻看评论之后发现这个解法,并尝试自己去按照该方式解决。
岁月悠悠,衰微只及肌肤;热忱抛却,颓废必致灵魂