需要用到一个栈,这个栈里保存的是nums数组的下标,下标所指向的数 从里到外单调递减。
我们把返回的结果命名为res, 那么它的长度为nums.size(), 默认每个元素都为-1
- 第一次遍历
1. 首先从左到右遍历一遍数组,每拿到一个数,就拿他和 栈顶所代表的数比较(栈顶只是个下标,我们要比较数字) 。
2. 如果栈顶代表的数字更小,说明 它的下一个更大的数 就是 当前的数。那么我们在结果数组里标记一下。并把栈顶的数除去,再看栈顶代表的数是否小于当前的数,重复操作,直到当前数是更小的为止。
3. 把 当前数的下标 推入栈。
4. 这样第一次遍历过后,栈的下标所指向的数从里到外单调递减。
- 再重新从左到右遍历一次
- 仍然重复第一次遍历中的1 2操作,但不再做操作 3。
把res返回。
也就是说,两次遍历中只有操作3是在第二次遍历中不需要做的,那么我们可以一次循环完成。
代码如下。
class Solution {
public:
vector<int> nextGreaterElements(vector<int>& nums) {
vector<int> res;
res.reserve(nums.size());
stack<int> st;
for(int i = 0; i < nums.size(); i++)
res.push_back(-1);
for(int i = 0; i < nums.size() * 2; i++){
int index = i % nums.size();
while(!st.empty()
&& nums[st.top()] < nums[index]){
res[st.top()] = nums[index];
st.pop();
}
if (i < nums.size())
st.push(index);
}
return res;
}
};