题目来源:https://leetcode.cn/problems/next-greater-element-ii/description/
C++题解:因为是循环数组,所以对数组进行了两次遍历,相当于循环。使用了栈,一个存放元素,一个存放索引,用来更新result。
class Solution {
public:
vector<int> nextGreaterElements(vector<int>& nums) {
int len = nums.size();
vector<int> result(len, -1);
stack<int> st, stind;
st.push(nums[0]);
stind.push(0);
for(int i = 1; i < len; i++) {
while(!st.empty() && nums[i] > st.top()) {
st.pop();
result[stind.top()] = nums[i];
stind.pop();
}
st.push(nums[i]);
stind.push(i);
}
for(int i = 0; i < len; i++) {
while(!st.empty() && nums[i] > st.top()) {
st.pop();
result[stind.top()] = nums[i];
stind.pop();
}
st.push(nums[i]);
stind.push(i);
}
return result;
}
};