hdu 4421 Bit Magic 2-sat

题意：a[i]数组可通过一定方式生成b[i][i]矩阵，问：给定b[i][i]矩阵，有没有满足要求的a[i]矩阵…
方法：枚举每一位数字上的数，建图，跑2-sat

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<queue>
#include<set>
using namespace std;
typedef long long ll;
const int N=505;
const int maxn=505;
int b[N][N];
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(),mark[i]=false;
    }
    void add(int x,int y){
        G[x].push_back(y);
        G[y].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;
    }
}A;
bool odd(int x){
    return x&1;
}
bool even(int x){
    return !odd(x);
}
bool a[N][N][32];
bool solve(int n){
    int x,y;
    for(int i=0;i<n;i++){
        if(b[i][i]!=0)return false;
        for(int j=i+1;j<n;j++)
            if(b[i][j]!=b[j][i])return false;
    }
    for(int k=0;k<31;k++){
        A.init(n);
        for(int i=0;i<n;i++){
            for(int j=i+1;j<n;j++){
                if(odd(i)&&odd(j)){
                    if(!a[i][j][k]){
                        A.add(2*i,2*j);
                    }
                }
                else if(even(i)&&even(j)){
                    if(a[i][j][k]){
                        A.add(2*i+1,2*j+1);
                    }
                }
                else{
                    if(a[i][j][k]){
                        A.add(2*i+1,2*j);
                        A.add(2*i,2*j+1);
                    }
                    else{
                        A.add(2*i,2*j);
                        A.add(2*i+1,2*j+1);
                    }
                }
            }
        }
        if(!A.solve())return false;
    }
    return true;
}
void work(){
    int n;
    while(~scanf("%d",&n)){        
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                scanf("%d",&b[i][j]);
                for(int k=0;k<31;k++)
                    if(b[i][j]&(1<<k))a[i][j][k]=true;
                    else a[i][j][k]=false;
            }
        }
        if(solve(n))puts("YES");        
        else puts("NO");
    }
}
int main(){
    //freopen("data.in","r",stdin);
    work();
    return 0;
}