• leetcode 【 Find Peak Element 】python 实现


    题目

    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.

    代码

    分别用递归和非递归两种方法实现。

    非递归版代码:oj测试通过 Runtime: 55 ms

     1 class Solution:
     2     # @param num, a list of integer
     3     # @return an integer
     4     def findPeakElement(self, num):
     5         if len(num) == 1 :
     6             return 0
     7         if len(num) == 2 :
     8             return [0,1][num[0] < num[1]]
     9         start = 0
    10         end = len(num)-1
    11         while start <= end:
    12             if start == end:
    13                 return start
    14             if start+1 == end:
    15                 return [start,end][num[start] < num[end]]
    16             mid = (start+end)/2
    17             if num[mid] < num[mid-1]:
    18                 #start = 0
    19                 end = mid -1
    20             elif num[mid] < num[mid+1]:
    21                 start = mid+1
    22                 #end = len(num)-1
    23             else:
    24                 return mid

    递归代码:oj测试通过 Runtime: 51 ms

     1 class Solution:
     2     # @param num, a list of integer
     3     # @return an integer
     4     def find(self,num,start,end):
     5         if start == end :
     6             return start
     7         if end == start + 1:
     8             if num[start] < num[end] :
     9                 return end
    10             else:
    11                 return start
    12         mid = (start+end)/2
    13         if num[mid] < num[mid-1] :
    14             return self.find(num, start, mid-1)
    15         if num[mid] < num[mid+1] :
    16             return self.find(num, mid, end)
    17         return mid
    18         
    19     def findPeakElement(self, num):
    20         if len(num) == 1:
    21             return 0
    22         return self.find(num, 0, len(num)-1)

    思路

    二分查找思路升级版。参照如下两篇日志的思路:

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

    http://blog.csdn.net/u010367506/article/details/41943309

    这道题在理解题意上需要注意就是默认这个数组的虚头和虚尾都是负无穷,并且这个数组不存在相等的两个元素,这样就一定能够找到题中描述的peak point.

  • 相关阅读:
    c++入门之初话结构体
    c++学习之字符串拼接
    数组赋值问题
    c++之sizeof的用法
    MySQL 创建一个简单的成绩管理系统
    KMP算法详解
    [Leetcode] Implement strstr
    [Leetcode] Multiply strings 字符串对应数字相乘
    [Leetcode] count and say 计数和说
    [Leetcode] Roman to integer 罗马数字转成整数
  • 原文地址:https://www.cnblogs.com/xbf9xbf/p/4242949.html
Copyright © 2020-2023  润新知