2021年度训练联盟热身训练赛第一场——Early Orders(单调栈)

原题链接
题意:
给出一个长度为n的序列和k,要求选出一个最短的子序列使得该子序列包含1~k所有数并且字典序最小。
思路:
一开始也想到单调栈了,然后用deque模拟一直没过。
大体思路就是:
如果当前栈为空的话,直接入栈;
如果不为空的话,考虑栈顶元素和当前元素的关系,如果栈顶元素大于当前元素并且栈顶元素在后面还有的话,就弹出栈顶元素,因为可以用后面的元素来替代这个元素。
最后倒着把序列输出就可以了。
代码:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll,ll>PLL;
typedef pair<int,int>PII;
typedef pair<double,double>PDD;
#define I_int ll
inline ll read()
{
    ll x=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9')
    {
        if(ch=='-')f=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
    {
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
#define read read()
#define closeSync ios::sync_with_stdio(0);cin.tie(0);cout.tie(0)
#define multiCase int T;cin>>T;for(int t=1;t<=T;t++)
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define repp(i,a,b) for(int i=(a);i<(b);i++)
#define per(i,a,b) for(int i=(a);i>=(b);i--)
#define perr(i,a,b) for(int i=(a);i>(b);i--)
ll ksm(ll a,ll b,ll p)
{
    ll res=1;
    while(b)
    {
        if(b&1)res=res*a%p;
        a=a*a%p;
        b>>=1;
    }
    return res;
}
#define PI acos(-1)
const double eps=1e-8;
const int maxn=1e6+100;
int a[maxn],n,k,vis[maxn],cnt[maxn];
stack<int>stk;
int main(){
    n=read,k=read;
    rep(i,1,n){
        a[i]=read;
        cnt[a[i]]++;
    }
    rep(i,1,n){
        cnt[a[i]]--;
        if(vis[a[i]]) continue;
        while(!stk.empty()&&cnt[stk.top()]&&stk.top()>a[i]){
            vis[stk.top()]=0;
            stk.pop();
        }
        stk.push(a[i]);
        vis[a[i]]=1;
    }
    int idx=0;
    while(!stk.empty()){
        a[++idx]=stk.top();
        stk.pop();
    }
    for(int i=idx;i;i--){
        cout<<a[i]<<" ";
    }
    return 0;
}

posted @ 2021-03-07 20:25  OvO1  阅读(39)  评论(0编辑  收藏  举报