给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1。
示例 1:
输入: [1,2,1]
输出: [2,-1,2]
解释: 第一个 1 的下一个更大的数是 2;
数字 2 找不到下一个更大的数;
第二个 1 的下一个最大的数需要循环搜索,结果也是 2。
注意: 输入数组的长度不会超过 10000。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/next-greater-element-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
参考:https://mp.weixin.qq.com/s/X0kf4rB88oS-VOcI7OR6OA
思路 :
利用单调栈的思想来做,把待处理的元素从最后一个元素进行处理.
对每一个元素的处理方式都是类似的,如果栈顶的元素小于当前元素直接出栈,直到大于了当前元素,然后看看当前的栈是不是空的,如果是空的说明后面没有元素比他大,直接返回-1.如果不是空的就把 栈顶元素返回.以为 这个元素肯定比他大..
针对这道题因为这里是环形数组,所以我们要对上面的思路进行一个扩充,思路是想象扩充成原来的2倍,如果这样的话就额外增加了负担,所以我们用取余的思想来做
其实就是全算出来然后取前面的一段,后面的一段虽然被算出来并且被赋值,但是会被前面的一段所覆盖
class Solution {
public:
vector<int> nextGreaterElements(vector<int>& nums)
{
int n = nums.size();
vector<int> res(n); // 存放结果
stack<int> s;
// 假装这个数组长度翻倍了
for (int i = 2 * n - 1; i >= 0; i--) {
while (!s.empty() && s.top() <= nums[i % n])
s.pop();
res[i % n] = s.empty() ? -1 : s.top();
s.push(nums[i % n]);
}
return res;
}
};