POJ 3648 Wedding 2-sat输出一组解

题目：n对夫妻，参加婚礼，一对夫妻不能坐在桌子的同一侧，m对有性爱关系的*男*女不能坐在桌子的同一侧，新娘新郎必须坐上桌子，输出任意一种新娘对面坐的人的情况。

这个肯定就是2-sat问题，建图时注意建新郎和新娘的边，还有缩点构造新图的时候返向建边，因为当有一条边a->b的时候，表示选a必须选b那么这样我们在选择人的时候，就会有一连串的问题，但是我们直接选b但是他可以不一定选择a，这个新图是个对称的图，因为2-sat问题建图后又很强的对称性，一个最基本的关系是题目给你的m对关系，隐含的就是夫妻关系，他们是排斥的关系，所以要标记这个关系，topsort的时候夫妻一方被选另一方自动到桌子的对面。

#include<iostream>#include<cstring>#include<cstdio>#include<stack>#include<queue>#include<vector> using namespace std;const int nodes=1000;const int edges=100000;struct node{int v,next;}e[nodes];int n,m;int conflict[nodes],dis[nodes],in[nodes];int head[nodes],cnt;vector<int>ve[nodes];stack<int>s;int low[nodes],dfn[nodes],paint[nodes],instack[nodes],index,col;void Init(){while(!s.empty()) s.pop();for(int i=0;i<=nodes;i++) ve[i].clear();memset(dfn,0,sizeof(dfn));memset(low,0,sizeof(low));memset(paint,0,sizeof(paint));memset(instack,0,sizeof(instack));memset(head,-1,sizeof(head));index=col=1;cnt=0;}void add(int a,int b){e[cnt].v=b;e[cnt].next=head[a];head[a]=cnt++;}int MIN(int a,int b){if(a<b) return a;return b;}void tarjan(int u){dfn[u]=low[u]=index++; instack[u]=1;s.push(u);for(int i=head[u];i+1;i=e[i].next){int v=e[i].v;if(!dfn[v]){tarjan(v);low[u]=MIN(low[u],low[v]);}else if(instack[v]){low[u]=MIN(low[u],dfn[v]);}}if(dfn[u]==low[u]){int k=s.top();while(k!=u){s.pop();paint[k]=col;instack[k]=0;k=s.top();}s.pop();paint[u]=col;instack[u]=0;col++;}}void Input(){for(int i=1;i<=m;i++){int a,b;int aa,bb;char ch1,ch2;scanf("%d%c %d%c",&a,&ch1,&b,&ch2);if(ch1=='h'){a=a+n;aa=a-n;}else{aa=a+n;}if(ch2=='h'){b=b+n;bb=b-n;}else {bb=b+n;}add(a,bb);add(b,aa);}add(0,n);}void topsort(){queue<int>q;while(!q.empty()) q.pop();for(int i=0;i<2*n;i++){if(!in[i])q.push(i);}while(!q.empty()){int u=q.front();q.pop();if(!dis[u]){dis[u]=1;dis[conflict[u]]=2;}for(int i=0;i<ve[u].size();i++){int v=ve[u][i];if(--in[v]==0){q.push(v);}}}}void Output(){memset(conflict,0,sizeof(conflict));memset(dis,0,sizeof(dis));memset(in,0,sizeof(in));for(int i=0;i<n;i++){conflict[paint[i]]=paint[i+n];conflict[paint[i+n]]=paint[i];}for(int i=0;i<2*n;i++){for(int j=head[i];j+1;j=e[j].next){int v=e[j].v;if(paint[i]!=paint[v]){ve[paint[v]].push_back(paint[i]);in[paint[i]];}}}topsort();for(int i=1;i<n;i++){if(i!=1) printf(" ");if(dis[paint[i]]==dis[paint[0]]){printf("%dw",i);}else printf("%dh",i);}} void treatment(){int mark=1;for(int i=0;i<2*n;i++){if(!dfn[i]) tarjan(i);}col--;for(int i=0;i<n;i++){if(paint[i]==paint[i+n]){mark=0;break;}}if(!mark) printf("bad luck\n");else Output();}int main(){while(~scanf("%d%d",&n,&m),n+m){Init();Input();treatment();}return 0;}