HDU1824 2-sat

Let's go home

Time Limit: 10000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 1926 Accepted Submission(s): 812

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

Author

威士忌

代码：

//i表示留下，i+1表示回家，若a,b,c表示队员，则有(a*2,b*2+1),(a*2,c*2+1),(b*2+1,a*2),(c*2+1,a*2)
//(b*2,c*2),(c*2,b*2);(a*2+1,b*2),(a*2+1,c*2),(b*2+1,a*2),(c*2+1,a*2),(b*2+1,c*2+1),(c*2+1,b*2+1)
//建12条边。若a,b不能同时留下则有(a*2,b*2+1),(b*2,a*2+1),因为a,b都可以回去所以没有反向边。然后套模板。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
const int maxn=3003;
/********************** 2-sat模板 **********************/
struct Twosat{
    int n;
    vector<int> g[maxn*2];
    bool mark[maxn*2];
    int s[maxn*2],c;
    bool dfs(int x){
        if(mark[x^1]) return false;
        if(mark[x]) return true;
        mark[x]=true;
        s[c++]=x;
        for(int i=0;i<(int)g[x].size();i++)
            if(!dfs(g[x][i])) return false;
        return true;
    }
    void init(int n){
        this->n=n;
        for(int i=0;i<n*2;i++) g[i].clear();
        memset(mark,0,sizeof(mark));
    }
    void add_clause(int x,int xval,int y,int yval){//这个函数随题意变化
        x=x*2+xval;
        y=y*2+yval;
        g[x].push_back(y);
    }
    bool solve(){
        for(int i=0;i<n*2;i+=2)
            if(!mark[i]&&!mark[i+1]){
                c=0;
                if(!dfs(i)){
                    while(c>0) mark[s[--c]]=false;
                    if(!dfs(i+1)) return false;
                }
            }
        return true;
    }
};
/*********************** 2-sat模板 ************************/
int main(){
    int n,m,a,b,c;
    Twosat solver;
    while(scanf("%d%d",&n,&m)==2){
        solver.init(3*n);
        for(int i=0;i<n;i++){
            scanf("%d%d%d",&a,&b,&c);
            solver.add_clause(a,0,b,1);
            solver.add_clause(a,0,c,1);
            solver.add_clause(b,0,a,1);
            solver.add_clause(b,0,c,0);
            solver.add_clause(c,0,a,1);
            solver.add_clause(c,0,b,0);

            solver.add_clause(a,1,b,0);
            solver.add_clause(a,1,c,0);
            solver.add_clause(b,1,a,0);
            solver.add_clause(b,1,c,1);
            solver.add_clause(c,1,a,0);
            solver.add_clause(c,1,b,1);
        }
        for(int i=0;i<m;i++){
            scanf("%d%d",&a,&b);
            solver.add_clause(a,0,b,1);
            solver.add_clause(b,0,a,1);
        }
        if(solver.solve()) printf("yes
");
        else printf("no
");
    }
    return 0;
}