HDU1824-Let's go home(2-SAT)

来源：互联网发布：setscale java 编辑：程序博客网时间：2024/04/28 18:04

题目链接

题意：n个队伍，一个队伍3个人，要求如果队长不在那剩下两个队员必须在，如果剩下两个队员不在队长必须在。m种冲突关系，每种冲突关系中的两个人不能同时存在，问方案是否可行。

思路：利用矛盾关系建图，用2-SAT求解。注意每对冲突关系中，至少有一个要留下来。

代码：

#include <iostream>#include <cstdio>#include <cstring>#include <vector>#include <algorithm>using namespace std;const int MAXN = 5005;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 < 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^1].push_back(y);        g[y^1].push_back(x);    }    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;    }};TwoSAT solver;int main() {    int n, m;     while (scanf("%d%d", &n, &m) != EOF) {        solver.init(n * 3);         int a, b, c;        for (int i = 0; i < n; i++) {            scanf("%d%d%d", &a, &b, &c);            solver.add_clause(a, 1, b, 1);            solver.add_clause(a, 1, c, 1);        }        for (int i = 0; i < m; i++) {            scanf("%d%d", &a, &b);            solver.add_clause(a, 0, b, 0);         }        if (solver.solve())             printf("yes\n");        else            printf("no\n");    }    return 0;}

0 0