LG1155 双栈排序

题意

Tom最近在研究一个有趣的排序问题。如图所示,通过2个栈\(S_1\)\(S_2\),Tom希望借助以下4种操作实现将输入序列升序排序。

  • 操作a
    如果输入序列不为空,将第一个元素压入栈\(S_1\)
  • 操作b
    如果栈\(S_1\)不为空,将\(S_1\)栈顶元素弹出至输出序列
  • 操作c
    如果输入序列不为空,将第一个元素压入栈\(S_2\)
  • 操作d
    如果栈\(S_2\)不为空,将\(S_2\)栈顶元素弹出至输出序列

如果一个1-n的排列P可以通过一系列操作使得输出序列为1,2,…,(n-1),n,Tom就称PP是一个“可双栈排序排列”。例如(1,3,2,4)就是一个“可双栈排序序列”,而(2,3,4,1)不是。下图描述了一个将(1,3,2,4)排序的操作序列:<a,c,c,b,a,d,d,b>

当然,这样的操作序列有可能有几个,对于上例(1,3,2,4),<a,c,c,b,a,d,d,b>是另外一个可行的操作序列。Tom希望知道其中字典序最小的操作序列是什么。

分析

参照Rookie-CH的题解。

先考虑单栈情况:

定理:考虑对于任意两个数q[i]和q[j],它们不能压入同一个栈中的充要条件: 存在一个k,使得i<j<k且q[k]<q[i]<q[j]。
证明:
充分性:即如果满足上述条件,那么q[i]和q[j]一定不能压入同一个栈。

反证法:假设这两个数压入了同一个栈,那么压入q[k],因为q[k]比q[i]和q[j]都小,所以很显然,当q[k]没有被弹出的时候,另两个数也都不能被弹出(否则输出序列的数字顺序就不是1,2,3,…,n了)。而之后,无论其它的数字在什么时候被弹出,q[j]总是会在q[i]之前弹出,而q[j]>q[i],这显然是不正确的.

必要性:如果两个数不可以压入同一个栈,那么它们一定满足上述条件。

证明逆否命题:也就是"如果不满足上述条件,那么这两个数一定可以压入同一个栈。”不满足上述条件有两种情况:
情况1:对于任意i<j<k且q[i]<q[j],q[k]>q[i];(即对任意三个数,最小的总是在最前面)
情况2:对于任意i<j,q[i]>q[j]。
第一种情况:在q[k]被压入栈的时候,q[i]已经被弹出栈。那么,q[k]不会对q[j]产生任何影响(这里可能有点乱,因为看起来,q[j]<q[k]的时候是会有影响的,但实际上,这还需要另一个数r,满足j<k<r且 q[r]<q[j]<q[k],也就是证明充分性的时候所说的情况。而事实上我们现在并不考虑这个r,所以说q[k]对q[j]没有影响)。
第二种情况:可以发现这其实就是一个降序序列,所以所有数字都可以压入同一个栈。这样,原命题的逆否命题得证,所以原命题得证。

双栈情况:

判断是否有解和任意两个数能否压入同一个栈

  1. 对任意两个数q[i]和q[j],若存在一个k,使得i<j<k且q[k]<q[i]<q[j],则这两个数分别入s1栈和s2栈
  2. 将s1栈和s2栈中的数字分成两个顶点子集,同一顶点子集内不会出现任何连边,即不能压入同一个栈的所有数字被分到了两个栈中。任意两个不在同一栈的数字间连边。此时我们只考虑检查是否有解,所以只要花O(n)时间检查这个图是不是二分图,就可以得知是否有解了。

*(二分图是一种特殊类型的图:图中的顶点集被划分成X与Y两个子集,图中每条边的两个端点一定是一个属于X而另一个属于Y。二分图的匹配是求边的一个子集,该子集中的任意两条边都没有公共的端点。)

解题步骤:

  1. 检查数列中的数字是否满足进入同一个栈的条件,如果不满足则将边连上,构造二分图。
  2. 用dfs染色,把二分图染成1和2两种颜色,使得染色为1的结点被压入s1栈,染色为2结点被压入s2栈。具体:每次选取一个未染色的编号最小的结点,将它染色为1,并从它开始dfs染色,直到所有结点都被染色为止。这样,我们就得到了每个结点应该压入哪个栈中。注意dfs结束之后未被染色的数字是可以被放入同一个栈的,所以优先染色为1。如果发现某一个数字两次要染不同的颜色,则为不能输出。
  3. 模拟出栈操作。

时间复杂度\(O(n^2)\)

代码

#include<bits/stdc++.h>
#define rg register
#define il inline
#define co const
template<class T>il T read(){
	rg T data=0,w=1;
	rg char ch=getchar();
	while(!isdigit(ch)){
		if(ch=='-') w=-1;
		ch=getchar();
	}
	while(isdigit(ch))
		data=data*10+ch-'0',ch=getchar();
	return data*w;
}
template<class T>il T read(rg T&x){
	return x=read<T>();
}
typedef long long ll;
using namespace std;

co int N=1001;
bool edge[N][N];
int s[N],F[N],color[N];
stack<int> staA,staB;
void NoAnswer(){
	puts("0"),exit(0);
}
int n;
void dfs(int x,int c){
	color[x]=c;
	for(int i=1;i<=n;++i) if(edge[x][i]){
		if(color[i]==c) NoAnswer();
		if(!color[i]) dfs(i,3-c);
	}
}
int main(){
//	freopen(".in","r",stdin);
//	freopen(".out","w",stdout);.
	read(n);
	for(int i=1;i<=n;++i)
		read(s[i]);
	F[n+1]=0x3f3f3f3f;
	for(int i=n;i>=1;--i)
		F[i]=std::min(s[i],F[i+1]);
	for(int i=1;i<n;++i)
		for(int j=i+1;j<=n;++j) if(s[i]<s[j]&&F[j+1]<s[i])
			edge[i][j]=edge[j][i]=1;
	for(int i=1;i<=n;++i) if(!color[i])
		dfs(i,1);
	int aim=1;
	for(int i=1;i<=n;++i){
		if(color[i]==1) staA.push(s[i]),printf("a ");
		else staB.push(s[i]),printf("c ");
		while(staA.size()&&staA.top()==aim||staB.size()&&staB.top()==aim){
			if(staA.size()&&staA.top()==aim) staA.pop(),printf("b ");
			else staB.pop(),printf("d ");
			++aim;
		}
	}
	return 0;
}

posted on 2019-02-20 08:52  autoint  阅读(227)  评论(0编辑  收藏  举报

导航