hdu4421——2-sat

题意：给出了一个数组通过图示转换规则生成一个矩阵：

问题是：给出一个矩阵问是否存在一个数组可以通过图示转换得到。

可以把数组a中的每个数的每一个位看作一个顶点，由于给出了b，那我们也就知道了一系列的约束条件。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <stack>
#include <algorithm>
using namespace std;

const int maxn = 40000 + 10;
const int maxm = 1000000 + 10;
struct edge { int v, next; };
int martix[505][505], n;

struct Twosat
{
    int n;
    edge edges[maxm];
    int S[maxn], top, edgehead[maxn], tot, mark[maxn];

    void init(int n)
    {
        this->n = n;
        memset(edgehead, 0xff, sizeof(edgehead));
        memset(mark, 0, sizeof(mark));
        tot = 0;
    }

    void addedge(int x, int xval, int y, int yval)
    {
        x = x * 2 + xval; y = y * 2 + yval;
        edges[tot].v = y; edges[tot].next = edgehead[x];
        edgehead[x] = tot++;
    }

    bool dfs(int u)
    {
        if(mark[u ^ 1]) return false;
        if(mark[u]) return true;
        mark[u] = 1;
        S[top++] = u;
        for(int i = edgehead[u]; i != -1; i = edges[i].next)
            if(!dfs(edges[i].v)) return false;
        return true;
    }

    bool slove()
    {
        for(int i = 0; i < 2 * n; ++i)
        {
            if(mark[i])
            {
                mark[i] = 0;
                if(!dfs(i)) return false;
            }
        }
        for(int i = 0; i < 2 * n; i += 2)
        {
            if(!mark[i] && !mark[i + 1]);
            {
                top = 0;
                if(!dfs(i))
                {
                    while(top > 0) mark[S[--top]] = 0;
                    if(!dfs(i + 1)) return false;
                }
            }
        }
        return true;
    }
}T;

int main()
{
    freopen("in", "r", stdin);
    while(~scanf("%d", &n))
    {
        for(int i = 0; i < n; ++i) for(int j = 0; j < n; ++j) scanf("%d", &martix[i][j]);
        bool res = true;
        for(int i = 0; i < n && res; ++i)
        {
            if(martix[i][i] != 0) res = false;
            for(int j = i + 1; j < n && res; ++j)
                if(martix[i][j] != martix[j][i]) res = false;
        }
        if(!res)
        {
            printf("NO\n");
            continue;
        }
        for(int k = 0; k < 31 && res; ++k)
        {
            T.init(n);
            for(int i = 0; i < n; ++i)
            {
                for(int j = i + 1; j < n; ++j)
                {
                    if(i % 2 == 1 && j % 2 == 1)
                    {
                        if(martix[i][j] & (1 << k))
                        {
                            T.addedge(i, 1, j, 0);
                            T.addedge(j, 1, i, 0);
                        }
                        else
                        {
                            T.mark[2 * i + 1] = T.mark[2 * j + 1] = 1;
                            //T.addedge(i, 0, i, 1);
                            //T.addedge(j, 0, j, 1);
                        }
                    }
                    else if(i % 2 == 0 && j % 2 == 0)
                    {
                        if(martix[i][j] & (1 << k))
                        {
                            T.mark[2 * i] = T.mark[2 * j] = 1;
                            //T.addedge(i, 1, i, 0);
                            //T.addedge(j, 1, j, 0);
                        }
                        else
                        {
                            T.addedge(i, 0, j, 1);
                            T.addedge(j, 0, i, 1);
                        }
                    }
                    else
                    {
                        if(martix[i][j] & (1 << k))
                        {
                            T.addedge(i, 0, j, 1);
                            T.addedge(i, 1, j, 0);
                            T.addedge(j, 0, i, 1);
                            T.addedge(j, 1, i, 0);
                        }
                        else
                        {
                            T.addedge(i, 0, j, 0);
                            T.addedge(i, 1, j, 1);
                            T.addedge(j, 0, i, 0);
                            T.addedge(j, 1, i, 1);
                        }
                    }
                }
            }
            if(!T.slove()) res = false;
        }
        if(res) printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}
/*
2
0 4
4 0
3
0 1 24
1 0 86
24 86 0

YES
NO
*/