和 739.每日温度 一样,用单调栈来求解
这里的区别是,数组是可以循环查找的,
我一开始的想法有误区,我想的是走到了最后一个再重新置回 0 。但是这样就很难判断到底什么时候循环结束
这里使用的方法是 1. 循环的长度是两倍的数组长度, 2. 然后对 i 取模
class Solution {
public:
vector<int> nextGreaterElements(vector<int>& nums) {
// 区别就是在循环查找
// 如何循环?
stack<int> st;
vector<int> res(nums.size(), -1);
st.push(0);
for(int i = 1; i < nums.size() * 2; i++)
{
if(nums[i % nums.size()] <= nums[st.top()])
st.push(i % nums.size());
else
{
while(!st.empty() && nums[i % nums.size()] > nums[st.top()])
{
res[st.top()] = nums[i % nums.size()];
st.pop();
}
st.push(i % nums.size());
}
}
return res;
}
};