首页 > 代码库 > LeetCode 162. Find Peak Element 20170706
LeetCode 162. Find Peak Element 20170706
A peak element is an element that is greater than its neighbors.
Given an input array where num[i] ≠ num[i+1]
, find a peak element and return its index.
The array may contain multiple peaks, in that case return the index to any one of the peaks is fine.
You may imagine that num[-1] = num[n] = -∞
.
For example, in array [1, 2, 3, 1]
, 3 is a peak element and your function should return the index number 2.
题目大意:
一个峰值元素是指它比邻居都大
给一个数组,其中 num[i] ≠ num[i+1],找到它的峰值元素,并返回角标。num[-1] = num[n] = -∞
如果包含多个峰值,可以只返回其中的一个的角标。注意:算法应该是对数级的
解题思路:因为算法是对数级的,所以不能遍历,还是需要用二分法。首先,由于两端都是无穷小,所以必定存在峰值,而且峰值有可能有多个。可以先取数组的mid,如果mid比左边的数小,那么说明mid是处于从左边下降的阶段,那么mid左边必定存在峰值,如果mid比右边元素大,说明mid处于往右边上升的阶段,mid右边必定存在峰值。
class Solution(object):
def findPeakElement(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
length=len(nums)
if length==1:
return 0
left=0
right=length-1
while left<right-1:
mid=(left+right)/2
if nums[mid]<nums[mid+1]:
left=mid+1
else:
right=mid
if nums[left]>nums[right]:
return left
else :
return right
LeetCode 162. Find Peak Element 20170706