问题:
峰值元素是指其值大于左右相邻值的元素。给定一个输入数组 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。
说明:你的解法应该是 O(logN) 时间复杂度的。
思路及代码:
1. Python实现
class Solution:
def findPeakElement(self, nums: List[int]) -> int:
# 方法一:峰值一定存在(因为规定两头都为负无穷),题目要求返回任意一个,那么返回最大值一定符合
# return nums.index(max(nums))
# 方法二:一次遍历列表元素,判断条件:两边元素小于当前元素,找到一个直接返回,不需要接着判断,时间复杂度O(n)不符合
# 必须是字符串才能这样写:nums = '-inf' + nums + '-inf'
# nums.insert(0, float('-inf')) # 要加float,不然加入的-inf是字符串,而不是数值-inf,下面报错:int和str无法比较
# nums.append(float('-inf'))
# for i in range(1, len(nums)): # 因为峰值一定存在,那么for循环一定有返回值,所以循环之后不需要返回值
# if ((nums[i] > nums[i-1]) and (nums[i] > nums[i+1])):
# return i - 1 # 加入了一个nums[-1],因此,以前的索引值是现在的索引值减一
# 方法三:二分查找法。
# 若nums[mid] < nums[mid-1],则在mid左边一定有峰值;
# 若nums[mid] < nums[mid+1],则在mid右边一定有峰值。
# 若上述两个条件均不满足,那么当前nums[mid]就是一个峰值。
if len(nums) == 1: return 0
low = 0; high = len(nums) - 1
while high - low > 2:
mid = (low + high) // 2
if nums[mid] < nums[mid-1]:
high = mid
elif nums[mid] < nums[mid+1]:
low = mid
else:
return mid
return nums.index(max(nums[low], nums[high], nums[low+1]))
知识点:
1. 注意区分:字符串采用 '+' 来连接;数组用nums.append()、nums.insert()来添加元素。