Gym 101201F 2-SAT

题意:

题目链接:http://codeforces.com/gym/101201/attachments
给出一个方格,其中若干点上有灯,每盏灯可以选择在一行或者一列照亮以自己为中心长度为2*r+1的区间,但是要让同一行之间以及同一列之间的灯不能有照亮的重叠区域,问是否可行。

思路:

每盏灯有两种选择,2-SAT裸题。

代码:

#include 
using namespace std;

const int MAXN = 1005;

struct node {
    int x, y;
} a[MAXN];

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;
     }
} twoSAT;

int main() {
    //freopen("in.txt", "r", stdin);
    int n, r, m;
    scanf("%d%d%d", &n, &r, &m);
    for (int i = 0; i < m; i++) {
        scanf("%d%d", &a[i].x, &a[i].y);
    }
    twoSAT.init(m);
    for (int i = 0; i < m; i++) {
        for (int j = i + 1; j < m; j++) {
            if (abs(a[i].x - a[j].x) < 2 * r && a[i].y == a[j].y) {
                twoSAT.add_clause(i, 0, j, 1);
                twoSAT.add_clause(j, 0, i, 1);
            }
            if (abs(a[i].y - a[j].y) < 2 * r && a[i].x == a[j].x) {
                twoSAT.add_clause(i, 1, j, 0);
                twoSAT.add_clause(j, 1, i, 0);
            }
        }
    }
    if (twoSAT.solve()) puts("YES");
    else puts("NO");
    return 0;
}

你可能感兴趣的:(2-SAT)