【LEETCODE】162-Find Peak Element [Python]

最新推荐文章于 2024-11-09 10:58:03 发布

Alice熹爱学习

最新推荐文章于 2024-11-09 10:58:03 发布

阅读量1.5k

点赞数

CC 4.0 BY-SA版权

分类专栏： LEETCODE 文章标签： LEETCODE PYTHON

本文链接：https://blog.youkuaiyun.com/aliceyangxi1987/article/details/50484982

LEETCODE 专栏收录该内容

137 篇文章

订阅专栏

本文介绍了一种使用二分查找法在给定数组中寻找峰值元素的方法。峰值元素定义为大于其邻居的元素。文章提供了一个Python实现示例，该示例在对数时间内找到并返回峰值元素的索引。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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.

click to show spoilers.

Note:

Your solution should be in logarithmic complexity.

题意：

一个峰值元素是指它比邻居都大

给一个数组，其中 num[i] ≠ num[i+1]，找到它的峰值元素，并返回角标

num[-1] = num[n] = -∞

如果包含多个峰值，可以只返回其中的一个的角标

注意：

算法应该是对数级的

思路：

用二分法，实现对数复杂度

是为了找极大值，所以，只有当num[mid]小于两边的时候，才会继续向两边寻找大的值

当num[mid]同时大于两边的时候，它就是极大值

参考：

http://bookshadow.com/weblog/2014/12/06/leetcode-find-peak-element/

Python:

class Solution(object):
    def findPeakElement(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        
        return self.helpsearch(nums,0,len(nums)-1)
    
    def helpsearch(self,nums,start,end):
        
        if start==end:
            return start
        if end-start==1:
            return [end,start][nums[start]>nums[end]]
        
        mid=(start+end)/2
        if nums[mid]<nums[mid-1]:
            return self.helpsearch(nums,start,mid-1)
        if nums[mid]<nums[mid+1]:
            return self.helpsearch(nums,mid+1,end)
        return mid