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[N];
int tot,tt,ans;
int head[N],dfs[N];  //dfs判断是否遍历
int inser[N],low[N],belong[N];  //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");
    }
}