poj 3648 Wedding

新郎和新娘分别做到桌子的两边,夫妻不能同时坐到一边,不正当关系的两个人也不能同时坐到新娘对面,输出坐到新娘对面的一组可行方案。

一对夫妻选择一个人坐到对面,很明显2-sat结构。第i对夫妻用2*i和2*i+1表示妻子和丈夫,对于每个不正当关系,把不正当关系中的a-b',b-a‘连边。然后逆序建图,拓扑排序染色,输出结果。

#include <iostream>
#include <cstdio>
#include <vector>
#include <cmath>
#include <stack>
#include <map>
#include <set>
#include <queue>
#include <cstring>
#include <algorithm>
using namespace std;
#define L(t) t<<1
#define R(t) t<<1|1
#define eps 1e-7
typedef long long LL;
const int maxn=5050;
const int maxm=100005;
const int MOD=1000000007;
int head[maxn],next[maxm],to[maxm],tot;
int dfn[maxn],low[maxn],stk[maxn],scc[maxn],cnt,top,id,f[maxn];
bool vis[maxn];
int color[maxn],ind[maxn];
vector<int> dag[maxn];
void addEdge(int u,int v)
{
    next[tot]=head[u];to[tot]=v;head[u]=tot++;
}
void init()
{
    tot=id=top=cnt=0;
    memset(head,-1,sizeof(head));
    memset(dfn,0,sizeof(dfn));
    memset(vis,0,sizeof(vis));
    memset(ind,0,sizeof(ind));
    memset(color,0,sizeof(color));
}
void tarjan(int v)
{
    dfn[v]=low[v]=++cnt;
    vis[v]=1;
    stk[++top]=v;
    int u;
    for(int i=head[v];i!=-1;i=next[i])
    {
        u=to[i];
        if(!dfn[u])
        {
            tarjan(u);
            low[v]=min(low[v],low[u]);
        }
        else if(vis[u])
        {
            low[v]=min(low[v],dfn[u]);
        }
    }
    if(dfn[v]==low[v])
    {
        id++;
        while(1)
        {
            u=stk[top--];
            scc[u]=id;
            vis[u]=0;
            if(u==v)
                break;
        }
    }
}
void BuildDag(int n)
{
    for(int u=0;u<2*n;u++)
    {
        for(int i=head[u];i!=-1;i=next[i])
        {
            int v=to[i];
            if(scc[u]!=scc[v])
            {
                dag[scc[v]].push_back(scc[u]);
                ind[scc[u]]++;
            }
        }
    }
}
void topsort()
{
    queue<int>q;
    for(int i=1;i<=id;i++)
    {
        if(ind[i]==0)
            q.push(i);
    }
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        if(!color[u])
            color[u]=1,color[f[u]]=2;
        int d=dag[u].size();
        for(int i=0;i<d;i++)
        {
            int v=dag[u][i];
            ind[v]--;
            if(!ind[v])
                q.push(v);
        }
    }
}
void solve(int n)
{
    for(int i=0;i<2*n;i++)
        if(!dfn[i])tarjan(i);
    for(int i=0;i<n;i++)
    {
        if(scc[2*i]==scc[2*i+1])
        {
//            cout<<scc[2*i]<<" "<<scc[2*i+1]<<endl;
            puts("bad luck");
            return;
        }
        else f[scc[2*i]]=scc[2*i+1],f[scc[2*i+1]]=scc[2*i];
    }
    BuildDag(n);
    topsort();
    for(int i=2;i<2*n;i+=2)
    {
        if(i!=2)
            printf(" ");
//        cout<<color[i]<<" "<<color[0]<<endl;;
        if(color[scc[i]]==color[scc[0]])
            printf("%dw",i/2);
        else
            printf("%dh",i/2);
    }
    printf("\n");
}
int main()
{
    //freopen("in","r",stdin);
    int n,m,x1,x2,y1,y2;
    char c1,c2;
    while(scanf("%d%d",&n,&m)&&(n+m))
    {
        int x,y;
        init();
        for(int i=0;i<=2*n;i++)
        dag[i].clear();
        for(int i=0;i<m;i++)
        {
            scanf("%d%c%d%c",&x,&c1,&y,&c2);
            if(c1=='h')
                x1=2*x,x2=2*x+1;
            else
                x1=2*x+1,x2=2*x;
            if(c2=='h')
                y1=2*y,y2=2*y+1;
            else
                y1=2*y+1,y2=2*y;
            addEdge(x2,y1);
            addEdge(y2,x1);
        }
        addEdge(0,1);
        solve(n);
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值