Perfect Flush--------------------------思维(单调栈)

最新推荐文章于 2024-01-25 19:59:16 发布

AKone123456

最新推荐文章于 2024-01-25 19:59:16 发布

阅读量236

点赞数

分类专栏：思维

本文链接：https://blog.csdn.net/qq_43690454/article/details/105950105

版权

思维专栏收录该内容

444 篇文章 1 订阅

订阅专栏

在这里插入图片描述

题意:
给定n个数和一个k，1<=a[i]<=k 让你找出k个数为1~k。输出字典序最小的k个数

解析:
利用单调栈，如果当前元素x <栈顶元素，则出栈。
事实并没有这么简单。
我们需要用两个数组分别记录每个数出现的次数和数是否在栈里面

设栈顶元素为y
如果x<y 且 cnt[y]=0 那么这个y在栈里面是固定的不能移动了。只能把x插在y后面了

如果x<y 且 cnt[y]!=0 那么这个y会被弹出栈，然后x继续和栈内元素比较。y被弹出去后，用 in[y]=0 标记一下 y已经不在栈里面了。

就这样遍历下去，字典序会最小


#include<bits/stdc++.h>
using namespace std;
const int N=2e5+1000; 
int n,k;
int in[N],cnt[N],w[N]; 
int a[N];
int main()
{
	cin>>n>>k;
	for(int i=1;i<=n;i++) 
	{
		cin>>a[i];
		cnt[a[i]]++;
	}
	stack<int> s;
	s.push(a[1]);cnt[a[1]]--;in[a[1]]=1;
	for(int i=2;i<=n;i++)
	{
		cnt[a[i]]--;
		if(in[a[i]]) continue;
		while(s.size()&&s.top()>a[i]&&cnt[s.top()])
		{
			in[s.top()]=0;
			s.pop(); 
		} 
		s.push(a[i]);
		in[a[i]]=1;
	}
	for(int i=1;i<=k;i++) w[i]=s.top(),s.pop();
	for(int i=k;i>=1;i--)
	{
		if(i==k) cout<<w[i];
		else cout<<" "<<w[i];
	 }
	cout<<endl;
}