poj3648 Wedding（2-sat）

最新推荐文章于 2019-03-03 16:16:00 发布

原创最新推荐文章于 2019-03-03 16:16:00 发布 · 202 阅读

0 ·

CC 4.0 BY-SA版权

其他oj 同时被 3 个专栏收录

336 篇文章

订阅专栏

---------图论---------

260 篇文章

订阅专栏

tarjan

38 篇文章

订阅专栏

本文介绍了一种使用图算法解决婚姻分配问题的方法。通过构建一个特殊的图模型，并利用Tarjan算法进行强连通分量的计算，确保每对夫妻不会被分配到同一组中，同时满足特定的约束条件。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

首先每个人一个标号，然后每个人分为0–和bride一侧 1–和groom一侧两个点。夫妻不能同侧，建边。有**的不能都是1，建边。然后规定groom必须为1，bride必须为0.

我本来写对了，然而对着一个正确的程序调了半天【再见】

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <stack>
#include <queue>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 100
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    if(ch=='w') return x<<1;if(ch=='h') return x<<1|1;
    return x*f;
}
int n,m,tot,id[N][2],h[N<<1],num=0,in[N<<1];//0--和bride一侧  1--和groom一侧
int dfn[N<<1],low[N<<1],dfnum,scc,bel[N<<1],col[N<<1],opp[N<<1];
char s[10];bool inq[N<<1];stack<int>qq;queue<int>q;
struct edge{
    int fr,to,next;
}data[100010];
inline void add(int x,int y){
    data[++num].to=y;data[num].next=h[x];h[x]=num;data[num].fr=x;
}
inline void tarjan(int x){
    low[x]=dfn[x]=++dfnum;qq.push(x);inq[x]=1;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;
        if(!dfn[y]) tarjan(y),low[x]=min(low[x],low[y]);
        else if(inq[y]) low[x]=min(low[x],dfn[y]);
    }if(low[x]==dfn[x]){
        ++scc;while(1){
            int y=qq.top();qq.pop();inq[y]=0;
            bel[y]=scc;if(y==x) break;
        }
    }
}
int main(){
//  freopen("a.in","r",stdin);
    while(1){
        n=read();m=read();if(!n&&!m) break;tot=-1;num=0;scc=0;dfnum=0;
        memset(dfn,0,sizeof(dfn));memset(low,0,sizeof(low));
        memset(h,0,sizeof(h));memset(in,0,sizeof(in));memset(col,0,sizeof(col));
        for(int i=0;i<n*2;++i) id[i][0]=++tot,id[i][1]=++tot;
        for(int i=1;i<n;++i){
            add(id[i<<1][0],id[i<<1|1][1]),add(id[i<<1][1],id[i<<1|1][0]);
            add(id[i<<1|1][0],id[i<<1][1]),add(id[i<<1|1][1],id[i<<1][0]);
        }add(id[0][1],id[0][0]);add(id[1][0],id[1][1]);
        while(m--){
            int x=read(),y=read();
            add(id[x][1],id[y][0]);add(id[y][1],id[x][0]);
        }for(int i=0;i<=tot;++i) if(!dfn[i]) tarjan(i);bool flag=0;
        for(int i=0;i<n*2;++i)
            if(bel[id[i][0]]==bel[id[i][1]]){puts("bad luck");flag=1;break;}
        if(flag) continue;
        memset(h,0,sizeof(h));int num1=num;num=0;
        for(int i=1;i<=num1;++i){
            int x=data[i].fr,y=data[i].to;
            if(bel[x]!=bel[y]) add(bel[y],bel[x]),in[bel[x]]++;
        }for(int i=1;i<=scc;++i) if(!in[i]) q.push(i);
        for(int i=0;i<n*2;++i) opp[bel[id[i][0]]]=bel[id[i][1]],opp[bel[id[i][1]]]=bel[id[i][0]];
        while(!q.empty()){
            int x=q.front();q.pop();
            for(int i=h[x];i;i=data[i].next){
                int y=data[i].to;if(--in[y]==0) q.push(y);
            }if(col[x]) continue;col[x]=1;col[opp[x]]=2;
        }for(int i=2;i<n*2;++i)
            if(col[bel[id[i][0]]]==1) printf("%d%c ",i/2,i%2?'h':'w');puts("");
    }return 0;
}