链接
https://leetcode-cn.com/problems/next-greater-element-ii/
耗时
解题:15 min
题解:9 min
题意
给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1。
注意: 输入数组的长度不会超过 10000。
思路
第一次遍历数组中每个元素,将其放入堆栈,在入栈之前,判断,如果当前元素大于栈顶元素,则当前元素即是栈顶元素的下一个更大的数,将结果数组中栈顶元素对应的位置变为当前元素,并出栈栈顶元素,直到栈顶元素不再大于当前元素,入栈当前元素。
因为是循环数组,所以需要第二次遍历数组中每个元素,这一次不再入栈,其余与第一次相同。
每个元素入栈出栈一次。
时间复杂度: O ( n ) O(n) O(n)
AC代码
class Solution {
public:
vector<int> nextGreaterElements(vector<int>& nums) {
int n = nums.size();
vector<int> ans(n, -1);
stack<int> st;
for(int i = 0; i < n; ++i) {
while(!st.empty() && nums[i] > nums[st.top()]) {
ans[st.top()] = nums[i];
st.pop();
}
st.push(i);
}
for(int i = 0; i < n; ++i) {
while(!st.empty() && nums[i] > nums[st.top()]) {
ans[st.top()] = nums[i];
st.pop();
}
}
return ans;
}
};