Leetcode每日一题——503. 下一个更大元素 II。单调栈

题目链接:

力扣

题目描述:

给定一个循环数组 nums ( nums[nums.length - 1] 的下一个元素是 nums[0] ),返回 nums 中每个元素的 下一个更大元素 。

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

示例 1:

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

输入: nums = [1,2,3,4,3]
输出: [2,3,4,-1,4]
 

提示:

1 <= nums.length <= 104
-109 <= nums[i] <= 109

解题思路:

单调栈解法:适用于一维数组,要寻找任一个元素的右边或者左边第一个比自己大或者小的元素的位置,此时我们就要想到可以用单调栈了。

        设计数据结构:单调栈,栈中元素(栈存的是下标,这里的元素指的是下标对应的元素)从栈底到栈顶递增

        设计算法:遍历数组(因为是循环的,所以这里直接扩充数组为原来的2倍)栈不空且遍历的元素大于栈顶元素,则res[栈顶元素%len]=当前遍历元素,直至栈顶元素大于当前遍历的元素为止,接着再让当前遍历元素入栈

代码如下:

stack = []#单调栈
        res = [-1] * len(nums)
        len1 = len(nums)
        nums = [j for i in range(2) for j in nums]#扩充nums,便于循环遍历
        for i in range(len(nums)):
            while len(stack) > 0 and nums[i] > nums[stack[-1]]:
                res[stack[-1] % len1] = nums[i]
                stack.pop()
            stack.append(i)
        return res#最后就算栈没空也不用管,因为此时栈内元素对应res里的值就是我们初始化的-1

你可能感兴趣的:(栈的应用,leetcode,算法,python)