2021年度训练联盟热身训练赛第一场 E题 Early Orders(栈模拟)

题意:
给出一个数组,只包含 1 1 1 k k k 的数,且每个数至少出现一次,求一个子序列,满足 1 1 1 k k k 的全排列。

题解:

用栈去模拟,当某个数 x x x 不在栈中,则一直与栈顶比较,直到 x x x 大于栈顶或栈顶元素已经全部出现完,

x x x 已经出现在栈中,那么直接跳过,不用考虑,因为此时栈中 x x x 后面的元素要么是单调递增的,要么存在某个数已经全部出现完,所以不需要再考虑。

代码:

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<map>
#include<stack>
#include<set>
#include<ctime>
#define iss ios::sync_with_stdio(false)
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
typedef pair<int,int>pii;
const int MAXN=2e5+50;
const int mod=1004535809;
const int inf=0x3f3f3f3f;
int vis[MAXN];
int a[MAXN];
int flag[MAXN];
std::vector<int> v;
int main()
{
    
    
    int n,k;
    cin>>n>>k;
    for(int i=1;i<=n;i++)
    {
    
    
        cin>>a[i];
        vis[a[i]]++;
    }
    stack<int>st;
    for(int i=1;i<=n;i++)
    {
    
    
        int x=a[i];
        vis[x]--;
        if(flag[x])
        {
    
    
            continue;
        }
        else
        {
    
    
            while(!st.empty())
            {
    
    
                if(x<st.top()&&vis[st.top()])
                {
    
    
                    flag[st.top()]=0;
                    st.pop();
                }
                else break;
            }
            st.push(x);
            flag[x]=1;
        }
    }
    while(!st.empty())
    {
    
    
        v.push_back(st.top());
        st.pop();
    }
    for(int i=v.size()-1;i>=0;i--)
    {
    
    
        printf("%d ",v[i]);
    }
}

猜你喜欢

转载自blog.csdn.net/weixin_45755679/article/details/114488509