poj3648(2-sat)

解题思路:有关2sat的部分详见我写的新的感想。大致来讲一下这道题的构图思路及意义吧。由于新娘不希望一对夫妻坐在对面,所以就是一个丈夫,妻子二选一的题目,非此即彼,所以可以用2sat来做。复杂度最高也O(NM),N为人数,M为关系数,近似于tarjan的复杂度。然后新娘也不希望有奸情的一对人坐在其对面。例如输入有奸情的为:2h 3w,则新娘的对面如果做了2h,则新娘对面也必须做3h,即连一条边2h->3h,同理还有3w->2w.由于前面的关系描述的都是新娘对面的关系,所以只能求新娘对面的座位情况。而新郎必须在新娘的对面,所以连一条从新娘到新郎的边。这样后面选择的时候,如果对面只有新郎自然不会有关系,但选了新娘,那新郎也一定会出现。所以不管怎么样,新郎是必须出现在选项中。然后剩下就是传统的2sat了

#include<cstdio>
#include<cstring>
#include<stack>
using namespace std;
const int MAXN = 2e4 + 5;
struct node
{
	int u,v,next;
}s[MAXN*2];
int n,m;
int head[100],cnt,dfn[100],low[100],p,k,ins[100],c,color[100];
int oppo[100],deg[100],jd[100],stck[100];
stack<int> q;
void add_edge(int u,int v)
{
	s[cnt].u=u;
	s[cnt].v=v;
	s[cnt].next=head[u];
	head[u]=cnt++;
}
void tarjan(int u)
{
	int tp;
	low[u]=dfn[u]=p++;
	stck[k++]=u;
	ins[u]=1;
	for(int i=head[u];i!=-1;i=s[i].next)
	{
		int v=s[i].v;
		if(!dfn[v])
		{
			tarjan(v);
			low[u]=min(low[u],low[v]);
		}else
		if(ins[v])
		{
			low[u]=min(dfn[v],low[u]);
		}
	}
	if(dfn[u]==low[u])
	{
		c++;
		do{
			tp=stck[--k];
			ins[tp]=0;
			color[tp]=c;	
		}while(u!=tp);
	}
}

void topo()
{
	while(!q.empty()) q.pop();
	for(int i=1;i<=c;i++)
	{
		if(deg[i]==0) q.push(i);
	}
	while(!q.empty())
	{
		int u=q.top();
		q.pop();
		if(!jd[u]) jd[u]=1,jd[oppo[u]]=-1;
		for(int i=head[u];i!=-1;i=s[i].next)
		{
			int v=s[i].v;
			deg[v]--;
			if(deg[v]==0)
			q.push(v);
		}
	}
}

void solve()
{
	p=1;
	k=0;
	c=0;
	memset(dfn,0,sizeof(dfn));
	memset(ins,0,sizeof(ins));
	memset(color,0,sizeof(color));
	memset(oppo,0,sizeof(oppo));
	for(int i=0;i<2*n;i++)
	{
		if(!dfn[i])
		tarjan(i);
	}
	//printf("*%d*",c);
	for(int i=0;i<n;i++)
	{
		if(color[i]==color[i+n])
		{
			printf("bad luck\n");
			return;
		}
		oppo[color[i]]=color[i+n];
		oppo[color[i+n]]=color[i];
	}
	memset(head,-1,sizeof(head));
	memset(deg,0,sizeof(deg));
	memset(jd,0,sizeof(jd));
	int num=cnt;
	for(int i=0;i<num;i++)
	{
		if(color[s[i].u]==color[s[i].v]) continue;
		add_edge(color[s[i].v],color[s[i].u]);
		deg[color[s[i].u]]++;
	}
	topo();
	for(int i=1;i<n;i++)
	{
		if(jd[color[i]]==1)
		printf("%dw ",i);
		else
		printf("%dh ",i);
	}
	printf("\n");
}

int main()
{
	//freopen("t.txt","r",stdin);
	int a1,a2;
	char c1,c2;
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		if(n==0&&m==0) break;
		cnt=0;
		memset(head,-1,sizeof(head));
		for(int i=0;i<m;i++)
		{
			scanf("%d%c%d%c",&a1,&c1,&a2,&c2);
			//printf("%d%c%d%c\n",a1,c1,a2,c2);
			if(c1=='w'&&c2=='w')
			{
				add_edge(a1+n,a2);
				add_edge(a2+n,a1);
			}else
			if(c1=='w'&&c2=='h')
			{
				add_edge(a1+n,a2+n);
				add_edge(a2,a1);
			}else
			if(c1=='h'&&c2=='w')
			{
				add_edge(a1,a2);
				add_edge(a2+n,a1+n);
			}else
			if(c1=='h'&&c2=='h')
			{
				add_edge(a1,a2+n);
				add_edge(a2,a1+n);
			}
		}
		add_edge(n,0);
		solve();		
	}
	return 0;
 } 

猜你喜欢

转载自blog.csdn.net/qq_39861441/article/details/87875223
今日推荐