题意:
给出一个数组,只包含 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]);
}
}