【LetMeFly】2454.下一个更大元素 IV:双单调栈
力扣题目链接:https://leetcode.cn/problems/next-greater-element-iv/
给你一个下标从 0 开始的非负整数数组 nums
。对于 nums
中每一个整数,你必须找到对应元素的 第二大 整数。
如果 nums[j]
满足以下条件,那么我们称它为 nums[i]
的 第二大 整数:
j > i
nums[j] > nums[i]
- 恰好存在 一个
k
满足i < k < j
且nums[k] > nums[i]
。
如果不存在 nums[j]
,那么第二大整数为 -1
。
- 比方说,数组
[1, 2, 4, 3]
中,1
的第二大整数是4
,2
的第二大整数是3
,3
和4
的第二大整数是-1
。
请你返回一个整数数组 answer
,其中 answer[i]
是 nums[i]
的第二大整数。
示例 1:
输入:nums = [2,4,0,9,6] 输出:[9,6,6,-1,-1] 解释: 下标为 0 处:2 的右边,4 是大于 2 的第一个整数,9 是第二个大于 2 的整数。 下标为 1 处:4 的右边,9 是大于 4 的第一个整数,6 是第二个大于 4 的整数。 下标为 2 处:0 的右边,9 是大于 0 的第一个整数,6 是第二个大于 0 的整数。 下标为 3 处:右边不存在大于 9 的整数,所以第二大整数为 -1 。 下标为 4 处:右边不存在大于 6 的整数,所以第二大整数为 -1 。 所以我们返回 [9,6,6,-1,-1] 。
示例 2:
输入:nums = [3,3] 输出:[-1,-1] 解释: 由于每个数右边都没有更大的数,所以我们返回 [-1,-1] 。
提示:
1 <= nums.length <= 105
0 <= nums[i] <= 109
方法一:双单调栈
思路
使用一个单调栈可以很方便地得出每个数的“第一个更大数”。但现在要求每个数的“第二个更大数”,怎么办呢?那就使用两个单调栈呗。
使用两个单调非递增栈st1和st2,首先将数存到st1中。如果st1中(栈顶)的数遇到了更大的数,就(不断)弹出(栈顶元素)并放入st2。
在st2中若栈顶元素遇到了更大的数a,那么a就是栈顶元素的“第二更大数”。
方法
遍历数组,先处理st2后处理st1,保证二者为单调(非递增)栈即可。
- 时间复杂度 O ( l e n ( n u m s ) ) O(len(nums)) O(len(nums))
- 空间复杂度 O ( l e n ( n u m s ) ) O(len(nums)) O(len(nums))
AC代码
C++
class Solution {
public:vector<int> secondGreaterElement(vector<int>& nums) {stack<int> st1, st2;vector<int> ans(nums.size(), -1);for (int i = 0; i < nums.size(); i++) {while (st2.size() && nums[st2.top()] < nums[i]) { // 第二次遇到“更大数”ans[st2.top()] = nums[i];st2.pop();}stack<int> temp; // temp的作用是保持st2的入栈顺序为大到小while (st1.size() && nums[st1.top()] < nums[i]) { // 第一次遇到更大数temp.push(st1.top());st1.pop();}st1.push(i);while (temp.size()) {st2.push(temp.top());temp.pop();}}return ans;}
};
Python
# from typing import Listclass Solution:def secondGreaterElement(self, nums: List[int]) -> List[int]:ans = [-1] * len(nums)st1 = []st2 = []for i in range(len(nums)):while st2 and nums[st2[-1]] < nums[i]:ans[st2[-1]] = nums[i]st2.pop()temp = []while st1 and nums[st1[-1]] < nums[i]:temp.append(st1[-1])st1.pop()st1.append(i)temp.reverse()st2 += tempreturn ans
同步发文于CSDN,原创不易,转载经作者同意后请附上原文链接哦~
Tisfy:https://letmefly.blog.csdn.net/article/details/134958992