NOIP2008 雙棧排序 twostack 題解

這道題的錯誤做法很多,但是實際在考場上,大多數人拿到了30分。錯誤做法卻能得滿分的也很多,正確的算法是基於二分圖的算法。注意,不是二分圖匹配!

分析條件,我們把問題抽象爲數學模型。設輸入序列爲S,考慮S[i],S[j]兩個元素不能進入同一個棧的條件.注意,這裏所說的"S[i],S[j]兩個元素不能進入同一個棧",不是說僅僅不能同時在一個棧中,而是自始至終不能進入一個棧,即如果有解,那麼S[i],S[j]一定進入過的棧不同.

結論P: S[i],S[j]兩個元素不能進入同一個棧 <=> 存在k,滿足i<j<k,使得S[k]<S[i]<S[j]. 證明略過,請參考sqybi.嘗試後可以發現結論P是正確的.

把每個元素按照輸入序列中的順序編號,看作一個圖中的每個頂點.這時,我們對所有的(i,j)滿足i<j,判斷是否滿足結論P,即S[i],S[j]兩個元素能否進入同一個棧.如果滿足P,則在i,j之間連接一條邊.

我們對圖染色,由於只有兩個棧,我們得到的圖必須是二分圖才能滿足條件.由於要求字典序最小,即儘量要進入棧1,我們按編號遞增的順序從每個未染色的頂點開始染色,相鄰的頂點染上不同的色,如果發生衝突,則是無解的.否則我們可以得到每個頂點顏色,即應該進入的棧.

接下來就是輸出序列了,知道了每個元素的決策,直接模擬了.

在判斷數對(i,j)是否滿足P時,枚舉檢查是否存在k的時間複雜度是O(n),則總的時間複雜度是O(n^3),對於n=1000是太大了.這原因在於過多得枚舉了k,我們可以用動態規劃把枚舉k變爲O(1)的算法.

設F[i]爲Min{S[i],S[i+1],S[i+2]..S[n-1],S[n]},狀態轉移方程爲F[i]=Min{ S[i] , F[i+1] }.邊界爲F[N+1]=極大的值.

判斷數對(i,j)是否滿足P,只需判斷(S[i]<S[j] 並且 F[j+1]<S[i])即可.時間複雜度爲O(n^2).

參考資料:sqybi的題解

/* 
 * Problem: NOIP2008 twostack
 * Author: Guo Jiabao
 * Time: 2008.12.9 21:22:52
 * State: Solved 
*/
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>

using namespace std;

const int MAXN=1002;
const int INF=0x7FFFFFFF;

class tStack
{
    private:
        int top;
        int S[MAXN];
    public:
        tStack() : top(0) {}
        void ins(int k)
        {
            S[++top]=k;
        }
        int tp()
        {
            return S[top];
        }
        void pop()
        {
            top--;
        }
};

int S[MAXN],F[MAXN],bel[MAXN];
bool adjm[MAXN][MAXN];
int N,top1,top2;
tStack T[3];

void init()
{
    int i;
    freopen("twostack.in","r",stdin);
    freopen("twostack.out","w",stdout);
    scanf("%d",&N);
    for (i=1;i<=N;i++)
    {
        scanf("%d",&S[i]);
    }
}

void noanswer()
{
    printf("0");
    exit(0);
}

void color(int i,int c)
{
    bel[i]=c;
    int j;
    for (j=1;j<=N;j++)
    {
        if (adjm[i][j])
        {
            if (bel[j]==c) //conflict : not a bipartite graph
            {
                noanswer();
            }
            if (!bel[j])
            {
                color(j,3-c); // color the opposite color 1<->2
            }
        }
    }
}

void dye()
{
    int i,j;
    F[N+1]=INF;
    for (i=N;i>=1;i--)
    {
        F[i]=S[i];
        if (F[i+1]<F[i])
            F[i]=F[i+1];
    }
    for (i=1;i<=N-1;i++)
    {
        for (j=i+1;j<=N;j++)
        {
            if (S[i]<S[j] && F[j+1]<S[i])
            {
                adjm[i][j]=adjm[j][i]=true;
            }
        }
    }
    for (i=1;i<=N;i++)
    {
        if (!bel[i])
        {
            color(i,1);
        }
    }
}

void solve()
{
    int i,should=1,s;
    for (i=1;i<=N;i++)
    {
        s=bel[i];
        if (s==1)
        {
            T[1].ins(S[i]);
            printf("a ");
        }
        else
        {
            T[2].ins(S[i]);
            printf("c ");
        }
        while (T[1].tp()==should || T[2].tp()==should)
        {
            if (T[1].tp()==should)
            {
                T[1].pop();
                printf("b");
                if (should!=N)
                    printf(" ");
                should++;
            }
            else
            {
                T[2].pop();
                printf("d");
                if (should!=N)
                    printf(" ");
                should++;
            }
        }
    }
}

int main()
{
    init();
    dye();
    solve();
    return 0;
}

相關日誌