洛谷 P1155 双栈排序

题目传送门

解题思路:

首先考虑只有一个栈的时候如何解决这个问题。

就是对于一对位置 (i, j)是否能共存三个位置 i<j<k 存在pk​<pi​<pj​ 是不可行的，因为pk​ 需要在 pi​ 与pj​ 之前出栈，但 pi​ 又需要在 pj​ 之前出栈，那么这就会产生矛盾。

我们就将i和j连一条边,如果找到一对i和j不能连边,构不成二分图,输出0;否则贪心模拟做题过程.

AC代码:

#include<cstdio>
#include<iostream>
#include<cstring>
#include<queue>
#include<vector>
#include<stack>
#include<cstdlib>
#define INF 0x7f7f7f7f

using namespace std;

int n,a[1005],f[1005],color[1005];
vector<int> e[1005];
stack<int> s1,s2;

inline void bfs(int s) {
    queue<int> q;
    q.push(s);
    color[s] = 1;
    while(!q.empty()) {
        int now = q.front();
        for(int i = 0;i < e[now].size(); i++) {
            int next = e[now][i];
            if(color[next] == -1) {
                color[next]= color[now] ^ 1;
                q.push(next); 
            }
            else if(color[next] != (color[now] ^ 1)) {
                printf("0");
                exit(0);
                return ; 
            }
        }
        q.pop();
    }
}

int main()
{ 
    memset(color,-1,sizeof(color));
    scanf("%d",&n);
    for(int i = 1;i <= n; i++) 
        scanf("%d",&a[i]);
    f[n+1] = INF;
    for(int i = n;i >= 1; i--) 
        f[i] = min(f[i+1],a[i]);
    for(int i = 1;i <= n; i++)
        for(int j = i + 1;j <= n; j++)
            if(a[i] > f[j+1] && a[i] < a[j])
                e[i].push_back(j),e[j].push_back(i);
    for(int i = 1;i <= n; i++)
        if(color[i] == -1)
            bfs(i);
    int cnt = 1;
    for(int i = 1;i <= n; i++) {
        if(color[i] == 1) {
            s1.push(a[i]);
            printf("a ");
        }
        else {
            s2.push(a[i]);
            printf("c ");
        }
        while((!s1.empty() && s1.top() == cnt) || (!s2.empty() && s2.top() == cnt)) {
            if(!s1.empty() && s1.top() == cnt) {
                s1.pop();
                printf("b ");
            }
            else  {
                s2.pop();
                printf("d ");
            }
        ++cnt;
        }
    }
    return 0;
}

//NOIP提高2008 T4