洛谷 P1155 双栈排序

题目描述
Tom最近在研究一个有趣的排序问题。如图所示，通过2个栈S1和S2，Tom希望借助以下4种操作实现将输入序列升序排序。

操作a
如果输入序列不为空，将第一个元素压入栈S1
操作b
如果栈S1不为空，将S1栈顶元素弹出至输出序列
操作c
如果输入序列不为空，将第一个元素压入栈S2
操作d
如果栈S2不为空，将S2栈顶元素弹出至输出序列
如果一个1~n的排列P可以通过一系列操作使得输出序列为1，2，…，(n-1)，n，Tom就称P是一个“可双栈排序排列”。例如(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希望知道其中字典序最小的操作序列是什么。
输入输出格式
输入格式：

输入文件twostack.in的第一行是一个整数n。
第二行有n个用空格隔开的正整数，构成一个1~n的排列。

输出格式：

输出文件twostack.out共一行，如果输入的排列不是“可双栈排序排列”，输出数字0；否则输出字典序最小的操作序列，每两个操作之间用空格隔开，行尾没有空格。

输入输出样例
输入样例#1：
【输入样例1】
4
1 3 2 4
【输入样例2】
4
2 3 4 1
【输入样例3】
3
2 3 1

输出样例#1：
【输出样例1】
a b a a b b a b
【输出样例2】
0
【输出样例3】
a c a b b d
说明
30%的数据满足： n<=10
50%的数据满足： n<=50
100%的数据满足： n<=1000

---

并不会做，上链接

---

#include<algorithm>
#include<iostream>
#include<cstdio>
#include<stack>
using namespace std;
const int N=1005;
stack<int>stka,stkb;
int n,aim,a[N],f[N],color[N];
bool flg,b[N][N];
void dfs(int u,int val)
{
    color[u]=val;
    for(int i=1;i<=n;i++)
        if(b[u][i])
        {
            if(color[i]==val)
                flg=1;
            else if(!color[i])
                dfs(i,3-val);
        }
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    f[n+1]=1e9+7;
    for(int i=n;i>=1;i--)
        f[i]=min(a[i],f[i+1]);
    for(int i=1;i<=n;i++)
        for(int j=i+1;j<=n;j++)
            if(a[i]<a[j]&&f[j+1]<a[i])
                b[i][j]=b[j][i]=1;
    for(int i=1;i<=n;i++)
        if(!color[i])
            dfs(i,1);
    if(flg)
    {
        printf("0\n");
        return 0;
    }
    aim=1;
    for(int i=1;i<=n;i++)
    {
        if(color[i]==1)
        {
            stka.push(a[i]);
            printf("a ");
        }
        else
        {
            stkb.push(a[i]);
            printf("c ");
        }
        while((!stka.empty()&&stka.top()==aim)||(!stkb.empty()&&stkb.top()==aim))
        {
            if(!stka.empty()&&stka.top()==aim)
            {
                stka.pop();
                printf("b ");
            }
            else
            {
                stkb.pop();
                printf("d ");
            }
            aim++;
        }
    }
    return 0;
}