C101【模板】单调栈 P5788 单调栈

视频链接:C101【模板】单调栈 P5788 单调栈_哔哩哔哩_bilibili

 

 

C15【模板】单调队列 滑动窗口最值 - 董晓 - 博客园 (cnblogs.com)

Luogu P5788 【模板】单调栈

// 单调栈 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]);
}

 

posted @ 2024-03-15 10:55  董晓  阅读(346)  评论(0编辑  收藏  举报