Gym - 101201F （2-SAT模板题）

题意：有N个灯，必须横着或竖着照r格，但任意一个格子不能同时被一行或一列的两个灯照到。
典型的2-SAT。列为真，行为假。因为列和行必须选一个。
xjb建建边就OK了

---

#include <iostream>
#include <cstring>
#include <string>
#include <algorithm>
#include <cstdio>
#include <cmath>
#include <map>
#include <set>
#include <queue>
#include <vector>
#define mod  1000000007
#define INF  0x3f3f3f3f
#define fuck() (cout << "----------------------------------------" << endl)
using namespace std;
const int maxn = 2000000 + 5;
int n ,r , l;
struct node
{
    int x ,y ;
    node(int xx = 0, int yy = 0):x(xx), y(yy){}
}pt[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 < 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;
bool check(int i, int j ,int ok)
{
    int x1 = pt[i].x, x2 = pt[j].x;
    int y1 = pt[i].y, y2 = pt[j].y;
    if(ok == 1)
    {
        if(x1 == x2 && abs(y1 - y2) <= 2 * r)
            return true;
        return false;
    }
    else
    {
        if(y1 == y2 && abs(x1 - x2) <= 2 * r)
            return true;
        return false;
    }
}
int main()
{
    scanf("%d%d%d",&n,&r,&l);
    twosat.init(l);
    for(int i = 0; i < l; i++)
        scanf("%d%d", &pt[i].x, &pt[i].y);
    for(int i = 0; i < l; i++)
        for(int j = i + 1; j < l; j++)
        {
            if(check(i,j,1))
                twosat.add_clause(i,1,j,1);
            if(check(i,j,0))
                twosat.add_clause(i,0,j,0);
        }
    if(twosat.solve()) printf("YES\n");
    else printf("NO\n");
}
/*
3 2 5
1 1
1 3
3 1
3 3
2 2
 */