POJ 3648 Wedding

2-SAT,直接选择新娘一侧的比较难做,所以处理的时候选择新郎一侧的,最后反着输出就可以。

A和B通奸的话,就建边 A->B'以及B->A’,表示 A在新郎一侧的话,B一定不在;B在新郎一侧的话,A一定不在。

然后再把新郎的mark标记为1,表示新郎一定选择。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<vector>
#include<algorithm>
using namespace std;

const int maxn=8000+5;
int N,m;
char s1[10],s2[10];

struct TwoSAT
{
    int n;
    vector<int> G[maxn*2];
    bool mark[maxn*2];
    int S[maxn*2],c;

    bool dfs(int x)
    {
        if(mark[x^1]) return false;
        if(mark[x]) return true;
        mark[x]=true;
        S[c++]=x;
        for(int i=0;i<G[x].size();i++)
            if(!dfs(G[x][i])) return false;
        return true;
    }

    void init(int n)
    {
        this->n=n;
        for(int i=0;i<n*2;i++) G[i].clear();
        memset(mark,0,sizeof mark);
    }

    void add_clause(int x,int y)
    {
        G[x].push_back(y^1);
        G[y].push_back(x^1);
    }

    bool solve()
    {
        for(int i=0;i<2*n;i+=2)
            if(!mark[i]&&!mark[i+1])
            {
                c=0;
                if(!dfs(i))
                {
                    while(c>0) mark[S[--c]]=false;
                    if(!dfs(i+1)) return false;
                }
            }
        return true;
    }

    //输出字典序最小的解
    void Printf()
    {
        for(int i=1;i<n;i++)
        {

            if(mark[2*i]) printf("%dw",i);
            else printf("%dh",i);

            if(i<n-1) printf(" ");
            else printf("\n");
        }
    }
};

int main()
{
    TwoSAT T;
    while(~scanf("%d%d",&N,&m))
    {
        if(N==0&&m==0) break;
        T.n=N; T.init(T.n);
        for(int i=0;i<m;i++)
        {
            int a=0,b=0;
            scanf("%s%s",s1,s2);

            for(int j=0;j<strlen(s1)-1;j++)
                a=a*10+s1[j]-'0';
            for(int j=0;j<strlen(s2)-1;j++)
                b=b*10+s2[j]-'0';

            if(s1[strlen(s1)-1]=='h') a=a*2;
            else a=a*2+1;

            if(s2[strlen(s2)-1]=='h') b=b*2;
            else b=b*2+1;

            T.add_clause(a,b);
        }
        T.mark[0]=1;
        if(T.solve()) T.Printf();
        else printf("bad luck\n");
    }
    return 0;
}

 

posted @ 2015-10-24 09:49  Fighting_Heart  阅读(177)  评论(0编辑  收藏  举报