tarjan 2-sat: party

最新推荐文章于 2023-02-17 10:17:07 发布

原创最新推荐文章于 2023-02-17 10:17:07 发布 · 942 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#insert #2010 #c #64bit #struct #output

ACM_图论专栏收录该内容

29 篇文章

订阅专栏

本文探讨了一个关于邀请夫妻参加聚会的图论问题，利用深度优先搜索（DFS）和Tarjan算法来解决是否存在一种邀请方案使得没有矛盾关系的人能同时出席。

Party Time Limit:1000MS Memory Limit:32768KB 64bit IO Format:%I64d & %I64u

Submit Status

Description

有n对夫妻被邀请参加一个聚会，因为场地的问题，每对夫妻中只有1人可以列席。在2n 个人中，某些人之间有着很大的矛盾（当然夫妻之间是没有矛盾的），有矛盾的2个人是不会同时出现在聚会上的。有没有可能会有n 个人同时列席？

Input

n：表示有n对夫妻被邀请 (n<= 1000)
m：表示有m 对矛盾关系 ( m < (n - 1) * (n -1))

在接下来的m行中，每行会有4个数字，分别是 A1,A2,C1,C2
A1,A2分别表示是夫妻的编号
C1,C2 表示是妻子还是丈夫，0表示妻子，1是丈夫
夫妻编号从 0 到 n -1

Output

如果存在一种情况则输出YES
否则输出 NO

Sample Input

Sample Output

YES

#include <cstdio>
#include <cstring>
#include <iostream>
#include <vector>
#include <cmath>
#include <stack>
#include <algorithm>

using namespace std;

stack<int> s;
bool vis[20100];
int low[20100];
int degree[20100];
int lastshow[20100];
int dfn[20100];
int cnt;
int n, m;
int Bcnt;
int id;
int index;
bool instack[20100];
int level[20100];

void init(){
    for(int i = 0;i <= 2*n+2; i ++){
        low[i] = 0;
        lastshow[i] = -1;
        dfn[i] = 0;
        level[i] = 0;
        instack[i] = false;
    }
    cnt = 0;
    index = 1;
    Bcnt = 0;
    while(!s.empty())
        s.pop();
}

struct edge{
    int to;
    int next;
}e[2000010];

void insert(int a, int b){
    e[cnt].to = b;
    e[cnt].next = lastshow[a];
    lastshow[a] = cnt++;
}

void dfs(int i){
    int j;
    low[i] = dfn[i] = index++;
    instack[i] = true;
    s.push(i);
    id = lastshow[i];
   // printf("x %d", id);
    while(id != -1){
        //printf("x %d", id);
        j = e[id].to;
        if(!dfn[j]){
           // printf("#%d#", dfn[j]);
            dfs(j);
            low[i] = min(low[i], low[j]);
        }
        else if(instack[j])
            low[i] = min(low[i], dfn[j]);
        id = e[id].next;
        //printf("d");
    }
    if(dfn[i] == low[i]){
        Bcnt ++;
        do{
            j = s.top();
            s.pop();
            level[j] = Bcnt;
            instack[j] = false;
        }
        while(j != i);
    }
}

bool solve(){
    bool flag = true;
    for(int i = 0; i <= n-1; i ++){
        if(level[2*i] == level[2*i+1]){
            flag = false;
            break;
        }
    }
    return flag;
}

int main(){
    int i, a1, a2, c1, c2;
    while(~scanf("%d %d", &n ,&m)){
        init();
        for(i = 1; i <= m; i ++){
            scanf("%d %d %d %d", &a1, &a2, &c1, &c2);
            insert(2 * a1 + c1, 2 * a2 + 1 - c2);
            insert(2 * a2 + c2, 2 * a1 + 1 - c1);
        }
        for(i = 0; i < 2*n; i ++){
            if(0 == dfn[i])
            dfs(i);
        }
//        for(i = 0; i < n; i ++){
//            printf("%d %d %d\n",dfn[i], low[i], level[i]);
//        }
        if(solve())
            printf("YES\n");
        else
            printf("NO\n");
    }
    return 0;
}