C101【模板】单调栈 P5788 单调栈
视频链接:C101【模板】单调栈 P5788 单调栈_哔哩哔哩_bilibili
C15【模板】单调队列 滑动窗口最值 - 董晓 - 博客园 (cnblogs.com)
// 单调栈 O(n) #include <iostream> #include <cstring> #include <algorithm> using namespace std; const int N=3000005; int n,a[N],ans[N],q[N]; int main(){ scanf("%d",&n); for(int i=1;i<=n;i++) scanf("%d",&a[i]); int top=0; //栈顶指针初值 for(int i=1; i<=n; i++){ while(top>0 && a[q[top]]<a[i]){ ans[q[top]]=i; //记录答案 top--; //栈顶出栈 } q[++top]=i; //栈顶入栈 } for(int i=1;i<=n;i++) printf("%d ",ans[i]); }
// 单调栈 O(n) #include <iostream> #include <cstring> #include <algorithm> #include <stack> using namespace std; const int N=3000005; int n,a[N],ans[N]; stack<int> q; //栈 int main(){ scanf("%d",&n); for(int i=1;i<=n;i++) scanf("%d",&a[i]); for(int i=1; i<=n; i++){ while(!q.empty() && a[q.top()]<a[i]){ ans[q.top()]=i; //记录答案 q.pop(); //栈顶出栈 } q.push(i); //栈顶入栈 } for(int i=1;i<=n;i++) printf("%d ",ans[i]); }