链接:https://leetcode-cn.com/problems/next-greater-element-ii/
给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1。
示例 1:
输入: [1,2,1] 输出: [2,-1,2] 解释: 第一个 1 的下一个更大的数是 2; 数字 2 找不到下一个更大的数; 第二个 1 的下一个最大的数需要循环搜索,结果也是 2。
分析:本题题意对每一个数A[i],找一个j>i,满足A[j]>A[i]且j离i最近。
思路1:单调栈可以用来求解位置i最近的比它大或者小的值。本题要求解的是右边第一个比它大的值,我们可以采用单调递减栈。注意题目要求的循环数组条件,我们可以将数组复制一遍拷贝在后面,当然也可以直接直接循环2*n次,配合取模n来做。
class Solution {
public:
struct Node{
int index, value;
};
vector<int> nextGreaterElements(vector<int>& nums) {
int n = nums.size(), i;
vector<int> ans(n, -1);
stack<Node> Stack;
for(i = 0; i < 2 * n; ++ i){
while(!Stack.empty() && Stack.top().value < nums[i % n]){
ans[Stack.top().index] = nums[i % n];
Stack.pop();
}
Stack.push(Node{i % n, nums[i % n]});
}
return ans;
}
};
思路2:动态规划优化。我们分析一下,朴素的枚举遍历算法的耗时:
A.如果存在一个下降区间,那么每次枚举时,都会因为经过这个下降区间而造成额外的开销。如序列 4 3 2 1 5 6 7 ,对4查询答案时,经过了下降区间[3,2,1],对3查询答案时,经过了下降区间[2,1],但是由于我们要找的答案是右边第一个比它大的数,因此绝不可能在下降区间上。
B.如果i<j,nums[i] < nums[j],dp[i]表示索引i处元素的右边第一个大于它的值的索引,那么很显然一定会满足dp[j]>=dp[i]。
如果i<j,nums[i] > nums[j],dp[i]表示索引i处元素的右边第一个大于它的值的索引,那么很显然一定会满足dp[j]<=dp[i]。