hdu1824 Let's go home (2-sat)

Problem Description

小时候，乡愁是一枚小小的邮票，我在这头，母亲在那头。

                        —— 余光中

集训是辛苦的，道路是坎坷的，休息还是必须的。经过一段时间的训练，lcy决定让大家回家放松一下，但是训练还是得照常进行，lcy想出了如下回家规定，每一个队（三人一队）或者队长留下或者其余两名队员同时留下；每一对队员，如果队员A留下，则队员B必须回家休息下，或者B留下，A回家。由于今年集训队人数突破往年同期最高记录，管理难度相当大，lcy也不知道自己的决定是否可行，所以这个难题就交给你了，呵呵，好处嘛~，免费**漂流一日。

 

Input

第一行有两个整数，T和M，1<=T<=1000表示队伍数，1<=M<=5000表示对数。

接下来有T行，每行三个整数，表示一个队的队员编号，第一个队员就是该队队长。

然后有M行，每行两个整数，表示一对队员的编号。

每个队员只属于一个队。队员编号从0开始。

 

Output

可行输出yes，否则输出no，以EOF为结束。

 

Sample Input

1 2
0 1 2
0 1
1 2

2 4
0 1 2
3 4 5
0 3
0 4
1 3
1 4

 

Sample Output

yes
no

 

Author

威士忌

思路：

2-SAT问题,拆点，回家i，不回家i+n*3；

建好队长跟队员之间的关系，每对队员之间的关系，

判断每个队员回不回家是否矛盾；

tarjan判断强连通；

代码：

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<stack>
using namespace std;
const int N=20000;
struct node{
int now,next,w;
}str
;
int tot,tt,ans;
int head
,dfs
; //dfs判断是否遍历
int inser
,low
,belong
; //inser判断是否在栈里，belong记录点属于哪个集合
stack<int>que; //先进先出
void relate(int x,int y) //邻接表建图
{
str[tot].now=y;
str[tot].next=head[x];
head[x]=tot++;
}
void tarjan(int x)
{
que.push(x);
dfs[x]=low[x]=tt++;
inser[x]=1;
for(int i=head[x];i!=-1;i=str[i].next)
{
int y=str[i].now;
if(dfs[y]==-1) //如果没有被遍历
{
tarjan(y);
low[x]=min(low[x],low[y]);
}
else if(inser[y]) //已经被遍历并且在栈里
low[x]=min(low[x],dfs[y]);
}
if(low[x]==dfs[x])
{
while(1)
{
int q=que.top();
que.pop();
inser[q]=0;
belong[q]=ans;
if(q==x)
{
break;
}
}
ans++;
}
}
int main()
{
int t,m;
while(~scanf("%d%d",&t,&m))
{
int n=3*t;
tot=0;tt=0;ans=0;
memset(dfs,-1,sizeof(dfs));
memset(head,-1,sizeof(head));
memset(inser,0,sizeof(inser));
for(int i=0;i<t;i++)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
relate(a+n,b);
relate(a+n,c);
relate(c+n,a);
relate(b+n,a);
}
for(int i=0;i<m;i++)
{
int a,b;
scanf("%d%d",&a,&b);
relate(a,b+n);
relate(b,a+n);
}
for(int i=0;i<2*n;i++)
{
if(dfs[i]==-1)
tarjan(i);
}
int flag=0;
for(int i=0;i<=n;i++)
{
if(i==n)
{
flag=1;
break;
}
if(belong[i]==belong[i+n])
break;
}
if(flag)
printf("yes\n");
else
printf("no\n");
}
}