HDU 1824 Let's go home (2-SAT判断)

题目链接:

http://acm.hdu.edu.cn/showproblem.php?pid=1824


【题目】

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



【思路】
每一个队伍中,要么队长留下,要么另外两个队员留下,这是一个矛盾对,可以把队长看成一个点,把两个队员也抽象成一个点,
把他们进行一个映射, 对于a,b,c, !a->b, !b->a, !c->a, 即f[a] = b, f[b]=a, f[c]=a,然后直接用2-SAT判断即可。




【代码】
#include<iostream> 
#include<cstdio>
#include<cstring>
using namespace std;

typedef long long int64;

const int MAXN = 3010;
const int VN = MAXN*2;
const int EN = VN*2;
int t, m;
int f[MAXN];

struct Edge{
    int v, next;
};

struct Graph{
public:
    void init(){
        size = 0;
        memset(head, -1, sizeof(head));
    }
    void addEdge(int u, int v){
        E[size].v = v;
        E[size].next = head[u];
        head[u] = size++;
    }

public:
    int head[VN];
    Edge E[EN];

private:
    int size; 
}g;

class Tow_Sat{
public:
    bool check(const Graph&g, const int n){
        scc(g, n);
        for(int i=0; i<n; ++i)
            if(belong[3*i] == belong[f[3*i]] ||
                belong[3*i+1] == belong[f[3*i+1]])
                return false;
        return true;

    }
private:
    int top, bcnt, idx;
    int DFN[VN];
    int low[VN];
    int belong[VN];
    int sta[VN];
    bool inStack[VN];

    void targan(const Graph&g, const int u){
        int v;
        DFN[u] = low[u] = ++idx;
        sta[top++] = u;
        inStack[u] = true;
        
        for(int e=g.head[u]; e!=-1; e=g.E[e].next){
            v = g.E[e].v;
            if(DFN[v] < 0){
                targan(g, v);
                low[u] = min(low[u], low[v]);

            }else if(inStack[v]){
                low[u] = min(low[u], DFN[v]);

            }
        }

        if(DFN[u] == low[u]){
            ++bcnt;
            do{
                v = sta[--top];
                inStack[v] = false;
                belong[v] = bcnt;
            }while(u != v);

        }
    }
    
    void scc(const Graph&g, int n){
        top = bcnt = idx = 0;
        memset(DFN, -1, sizeof(DFN));
        memset(inStack, 0, sizeof(inStack));
        for(int i=0; i<3*n; ++i)
            if(DFN[i] < 0) targan(g, i);
    }
}sat;

int main(){
    int a,b,c;

    while(~scanf("%d%d", &t, &m)){
    
        g.init();

        for(int i=0; i<t; ++i){
            scanf("%d%d%d", &a,&b,&c);
            g.addEdge(b, c);
            g.addEdge(c, b);
            f[a] = b;
            f[b] = f[c] = a;
        }
        bool flag = true;
        for(int i=0; i<m; ++i){
            scanf("%d%d", &a,&b);
            g.addEdge(a, f[b]);
            g.addEdge(b, f[a]);
        }
        if(flag && sat.check(g, t)) puts("yes");
        else puts("no");
    }
    return 0;
}







你可能感兴趣的:(home)