leetcode--下一个更大元素II--python

最新推荐文章于 2021-11-11 23:04:50 发布

原创最新推荐文章于 2021-11-11 23:04:50 发布 · 484 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #python #栈 #算法

leetcode 专栏收录该内容

27 篇文章

订阅专栏

本文介绍了一种解决循环数组中寻找每个元素下一个更大元素问题的高效算法。通过使用栈来存储未找到更大元素的位置，当遇到更大元素时，更新栈顶元素并出栈，直至栈空或无法找到更大元素。最后，未出栈元素均设为-1。此算法适用于编程竞赛及面试准备。

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

题目

题目详情

给定一个循环数组（最后一个元素的下一个元素是数组的第一个元素），输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序，这个数字之后的第一个比它更大的数，这意味着你应该循环地搜索它的下一个更大的数。如果不存在，则输出 -1。

示例

输入: [1,2,1]
输出: [2,-1,2]
解释: 第一个 1 的下一个更大的数是 2；
数字 2 找不到下一个更大的数； 
第二个 1 的下一个最大的数需要循环搜索，结果也是 2。

解题思路

思路

首先想到用栈的方法，存住暂时还没有找到比它大的数的位置
那么一旦找到那个比它大的数，就把暂存的位置给pop出来
并且修改pop出来的位置的值，直到运行完成
最后没有pop出来的位置都是没有找到比它们更大的数
遍历栈里没有pop的元素，并且赋值为-1

代码

class Solution(object):
	def nextGreaterElements(self, nums):
		if not nums:return []
		lengths = len(nums)
		nums = nums * 2
		static = [0]
		for i in range(1,len(nums)):         #遍历整个元素
			while static and nums[static[-1]] < nums[i]: #这是比较栈里是否有比当前数小的值
				nums[static[-1]] = nums[i]   #找到后出栈并且赋值
				static.pop()         
			static.append(i)         #如果还是没有找到比当前数小的值那么久入栈
		for j in static:
			nums[j] = -1
		return nums[:lengths]

if __name__ == "__main__":        #测试
	obj = Solution()
	obj.nextGreaterElements([3,2,1,0,1])   #输出结果[-1,3,3,1,3]

运行结果

在这里插入图片描述

最佳方案

172ms
他首先是找出了最大的一个元素的位置，以最大元素的位置作为起点，往前面比较

class Solution:
    def nextGreaterElements(self, nums):
        if not nums:
            return nums
        stack = list()
        res = [-1]*len(nums)
        max_index = nums.index(max(nums))     #找到最大的元素的位置
        for i in range(max_index, max_index-len(nums), -1):    #向前遍历，说明这些数都小于等于它
            while stack and stack[-1] <= nums[i]:  
                stack.pop()                  
            if stack:                      
                res[i] = stack[-1]
            stack.append(nums[i])
        return res