hdu 4421 Bit Magic(2-SAT)

题意：用一个矩阵给出n个数之间&、|、^的关系，问是否有一组数满足条件。

思路：数的范围是2^31-1，因此可以按位算，想到2-SAT的话建图其实也很简单。但是这题时限有点高，刚开始把所有位一起建图写超时了，后来看别人题解一位位计算，如果遇到矛盾，直接break掉，这样写1600ms+过掉了。2-SAT貌似可以用tarjan优化的，可惜我还不会……有时间看看论文吧……

 

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<queue>
#include<set>
#include<stack>
#include<cmath>
#include<vector>
#define inf 0x3f3f3f3f
#define Inf 0x3FFFFFFFFFFFFFFFLL
#define eps 1e-9
#define pi acos(-1.0)
using namespace std;
typedef long long ll;
const int maxn=500+10;
const int maxm=1000000+10;
struct Edge
{
    int v,next;
};
struct TwoSat
{
    int n;
    Edge edges[maxm<<1];
    int head[maxn<<1],S[maxn<<1],c,nEdge;
    bool mark[maxn<<1];
    void Init(int n)
    {
        this->n=n;
        memset(head,0xff,sizeof(head));
        memset(mark,0,sizeof(mark));
        nEdge=-1;
    }
    void AddEdge(int u,int v)
    {
        nEdge++;
        edges[nEdge].v=v;
        edges[nEdge].next=head[u];
        head[u]=nEdge;
    }
    void add_clause(int x,int xval,int y,int yval)
    {
        x=x*2+xval;
        y=y*2+yval;
        AddEdge(x^1,y);
        AddEdge(y^1,x);
    }
    bool dfs(int u)
    {
        if(mark[u^1]) return false;
        if(mark[u]) return true;
        S[c++]=u;
        mark[u]=true;
        for(int k=head[u];k!=-1;k=edges[k].next)
        {
            int v=edges[k].v;
            if(!dfs(v)) return false;
        }
        return true;
    }
    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 b[maxn][maxn],p[40];
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    for(int i=0;i<31;++i) p[i]=1<<i;
    int n;
    while(~scanf("%d",&n))
    {
        twosat.Init(n*31);
        for(int i=0;i<n;++i)
          for(int j=0;j<n;++j)
            scanf("%d",&b[i][j]);
        bool flag=true;
        for(int i=0;i<n;++i)
          if(b[i][i]) flag=false;
        if(!flag)
        {
            printf("NO\n");
            continue;
        }
        for(int k=0;k<31;++k)
        {
            twosat.Init(n);
            for(int i=0;i<n;++i)
             for(int j=0;j<n;++j)
             {
                 if(i==j) continue;
                 if(i%2==0&&j%2==0)
                 {
                     if(b[i][j]&p[k])
                     {
                         twosat.add_clause(i,1,j,1);
                         twosat.add_clause(i,0,j,1);
                         twosat.add_clause(i,1,j,0);
                     }
                     else twosat.add_clause(i,0,j,0);
                 }
                 else if(i%2==1&&j%2==1)
                 {
                     if(b[i][j]&p[k])
                        twosat.add_clause(i,1,j,1);
                     else
                     {
                         twosat.add_clause(i,0,j,1);
                         twosat.add_clause(i,1,j,0);
                         twosat.add_clause(i,0,j,0);
                     }
                 }
                 else
                 {
                     if(b[i][j]&p[k])
                     {
                         twosat.add_clause(i,1,j,1);
                         twosat.add_clause(i,0,j,0);
                     }
                     else
                     {
                         twosat.add_clause(i,0,j,1);
                         twosat.add_clause(i,1,j,0);
                     }
                 }
             }
            if(!twosat.solve())
            {
                flag=false;
                break;
            }
        }
        if(flag) printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}