峰值元素是指其值大于左右相邻值的元素。
给定一个输入数组 nums,其中 nums[i] ≠ nums[i+1],找到峰值元素并返回其索引。
数组可能包含多个峰值,在这种情况下,返回任何一个峰值所在位置即可。
你可以假设 nums[-1] = nums[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:
暴力查找。
遍历列表中的元素,如果某个元素满足左右两边的元素都小于该元素,就返回该元素的索引。此时应该注意列表两端的点是没有参与这个遍历过程的,所以如果遍历没能找到符合要求的峰值,此时利用[0,n-1][nums[0]<nums[n-1]] 这个语句来判定第0和第n-1个元素哪个大。如果第0个元素大,nums[0]<nums[n-1]为False,此时返回[0,n-1]中的第一个元素,反之,第n-1个元素大,nums[0]<nums[n-1]为True,此时返回[0,n-1]中的第二个元素。
class Solution(object):
def findPeakElement(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
n = len(nums)
for i in range(1,n-1):
if nums[i-1]<nums[i] and nums[i+1]<nums[i]:
return i
return [0,n-1][nums[0]<nums[n-1]]
方法2:二分查找
思路:
已知数组两端的点可以认为是负无穷,所以峰值一定是存在的。
选定mid,判断mid所在的点的趋势是什么,如果nums[mid]<nums[mid+1]—上升趋势,这个时候一定存在一个峰值在mid的右边,有可能是mid+1、mid+2等等,所以此时观察右边区域,更新low=mid+1;反之,处在下降趋势,则峰值在mid的左边,有可能是mid、mid-1等等,所以此时观察左边区域,更新high=mid。这样不断缩小查找区域,一定会找到low=high的点。
class Solution(object):
def findPeakElement(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
low = 0
high = len(nums)-1
while low!=high:
mid = (low+high)//2
if nums[mid]<nums[mid+1]:
low = mid+1
else:
high = mid
return low