Leetcode 162.寻找峰值
1 题目描述(Leetcode题目链接)
峰值元素是指其值大于左右相邻值的元素。给定一个输入数组 nums,其中 nums[i] ≠ nums[i+1],找到峰值元素并返回其索引。数组可能包含多个峰值,在这种情况下,返回任何一个峰值所在位置即可。你可以假设 nums[-1] = nums[n] = -∞。
输入: nums = [1,2,3,1]
输出: 2
解释: 3 是峰值元素,你的函数应该返回其索引 2。
输入: nums = [1,2,1,3,5,6,4]
输出: 1 或 5
解释: 你的函数可以返回索引 1,其峰值元素为 2;
或者返回索引 5, 其峰值元素为 6。
说明:你的解法应该是 O ( l o g N ) O(logN) O(logN) 时间复杂度的。
2 题解
题目要求的时间复杂度明显是个二分,那就来看这道题是怎么二分的,峰值就是比左右两边都大的一个值。二分每次取中间值,首先判断这个值是否是个峰值,是的话就返回这个中值就好。如果不是峰值,那就说明这个值的两侧有大于它的元素,我们要选取大于它的元素那一侧继续二分,因为这一侧一定存在峰值。比如1个例子 [ 1 , 2 , 3 , 4 , 5 , 7 , 6 ] [1,2,3,4,5,7,6] [1,2,3,4,5,7,6]第一次二分取到4,那么我们就选择右边这一侧二分。
class Solution:
def findPeakElement(self, nums: List[int]) -> int:
nums = [float("-inf")] + nums + [float("-inf")]
i, j = 0, len(nums) - 1
while i < j:
mid = (i+j)//2
if nums[mid] > nums[mid-1] and nums[mid] > nums[mid+1]:
return mid-1
if nums[mid] < nums[mid-1]:
j = mid - 1
else:
i = mid + 1
return i-1