503. 下一个更大元素 II
题目链接:503. 下一个更大元素 II
代码如下:
class Solution {
public:vector<int> nextGreaterElements(vector<int>& nums) {vector<int> res(nums.size(),-1);stack<int> sta;//单调栈sta.push(0);for(int i=1;i<nums.size()*2;i++){if(nums[i%nums.size()]<nums[sta.top()])//当前遍历的元素T[i]小于栈顶元素T[st.top()]的情况sta.push(i%nums.size());else if(nums[i%nums.size()]==nums[sta.top()])//当前遍历的元素T[i]等于栈顶元素T[st.top()]的情况sta.push(i%nums.size());else//当前遍历的元素T[i]大于栈顶元素T[st.top()]的情况{while(!sta.empty()&&nums[i%nums.size()]>nums[sta.top()]){res[sta.top()]=nums[i%nums.size()];sta.pop();}sta.push(i%nums.size());}}return res;}
};