HDU3410（单调栈） - 码农教程

题意描述

给定n个数字，求每个人左边的最大值和右边的最大值

思路

由于题目的数据范围较大，所以我们不能用暴力解法，可以考虑维护一个递减单调栈，可以使用两遍单调栈，先从左到右维护，然后再从右到左维护一遍。我们可以先用一个变量来记录栈顶，然后再pop()出去，这个变量即是下个栈顶右边或左边的最大值。

AC代码

#include<bits/stdc++.h>
#define x first
#define y second
#define IOS ios::sync_with_stdio(false);cin.tie(0);
using namespace std;
typedef unsigned long long ULL;
typedef pair<int,int> PII;
typedef pair<long,long> PLL;
typedef pair<char,char> PCC;
typedef long long LL;
const int N=50050;
const int M=150;
const int INF=0x3f3f3f3f;
const int MOD=998244353;
int Case=1;
int a[N],ans1[N],ans2[N];
void solve(){
    int n;scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    memset(ans1,0,sizeof ans1);
    memset(ans2,0,sizeof ans2);
    stack<int> s;
    for(int i=1;i<=n;i++){
        while(!s.empty() && a[s.top()]<a[i]){
            int t=s.top();s.pop();
            if(!s.empty()) ans1[s.top()]=t;
        }
        s.push(i);
    }
    while(!s.empty()){
        int t=s.top();s.pop();
        if(!s.empty()) ans1[s.top()]=t;
    }
    for(int i=n;i>=1;i--){
        while(!s.empty() && a[s.top()]<a[i]){
            int t=s.top();s.pop();
            if(!s.empty()) ans2[s.top()]=t;
        }
        s.push(i);
    }
    while(!s.empty()){
        int t=s.top();s.pop();
        if(!s.empty()) ans2[s.top()]=t;
    }
    printf("Case %d:n",Case++);
    for(int i=1;i<=n;i++){
        printf("%d %dn",ans2[i],ans1[i]);
    }
}
int main(){
    //IOS;
    int t;cin>>t;
    while(t--){
        solve();
    }
    return 0;
}