zoj 3656 bit magic

zoj 3656 2012 长春regional的题
比较明显的two-sat 是 katu_puzzle
那题目的升级版- -。做一些预处理就一样

了

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<vector>
#include<sstream>
#include<string>
#include<climits>
#include<stack>
#include<set>
#include<bitset>
#include<cmath>
#include<deque>
#include<map>
#include<queue>
#define iinf 2000000000
#define linf 1000000000000000000LL
#define dinf 1e200
#define eps 1e-11
#define all(v) (v).begin(),(v).end()
#define sz(x)  x.size()
#define pb push_back
#define mp make_pair
#define lng long long
#define sqr(a) ((a)*(a))
#define pii pair<int,int>
#define pll pair<lng,lng>
#define pss pair<string,string>
#define pdd pair<double,double>
#define X first
#define Y second
#define pi 3.14159265359
#define ff(i,xi,n) for(int i=xi;i<=(int)(n);++i)
#define ffd(i,xi,n) for(int i=xi;i>=(int)(n);--i)
#define ffl(i,r) for(int i=head[r];i!=-1;i=edge[i].next)
#define cc(i,j) memset(i,j,sizeof(i))
#define two(x)			((lng)1<<(x))
#define N 40000
#define M 1000000
#define lson l , mid , rt << 1
#define rson mid + 1 , r , rt << 1 | 1
#define Mod  n
#define Pmod(x) (x%Mod+Mod)%Mod
using namespace std;
typedef vector<int>  vi;
typedef vector<string>  vs;
typedef unsigned int uint;
typedef unsigned lng ulng;
template<class T> inline void checkmax(T &x,T y){if(x<y) x=y;}
template<class T> inline void checkmin(T &x,T y){if(x>y) x=y;}
template<class T> inline T Min(T x,T y){return (x>y?y:x);}
template<class T> inline T Max(T x,T y){return (x<y?y:x);}
template<class T> T gcd(T a,T  b){return (a%b)==0?b:gcd(b,a%b);}
template<class T> T lcm(T a,T b){return a*b/gcd(a,b);}
template<class T> T Abs(T a){return a>0?a:(-a);}
template<class T> inline T lowbit(T n){return (n^(n-1))&n;}
template<class T> inline int countbit(T n){return (n==0)?0:(1+countbit(n&(n-1)));}
template<class T> inline bool isPrimeNumber(T n)
{if(n<=1)return false;for (T i=2;i*i<=n;i++) if (n%i==0) return false;return true;}
template<class T> inline T Minmin(T a,T b,T c,T d){return Min(Min(a,b),Min(c,d));}
struct tarjan_seg
{
    int head[N],tot,id[N],pre[N],low[N],stack[N],top,n,Index,nn,newid,newhead[N],*cur,contain[N];
    struct segment
   {
        int v,next;
    }edge[1000000];
    void add(int u,int v)
    {
        edge[tot].v=v;
        edge[tot].next=cur[u];
        cur[u]=tot++;
    }
    void init()
    {
        cc(head,-1);
        cc(newhead,-1);
        tot=0;
        newid=0;
        cur=head;
        cc(pre,-1);
        cc(low,-1);
        top=0;
        cc(id,-1);
        Index=0;
    }
    void tarjan(int r,int p)
    {
        bool xx=false;
        pre[r]=low[r]=++Index;
        stack[++top]=r;
        ffl(i,r)
        {
            int v=edge[i].v;
            if(v==p&&xx==0)
            {
                xx=1;
                continue;
            }
            if(pre[v]==-1)
            {
                tarjan(v,r);
                checkmin(low[r],low[v]);
            }
            else
            if(id[v]==-1)
            checkmin(low[r],pre[v]);
        }
        if(pre[r]==low[r])
        {
            int v;
            ++newid;
            do
            {
                v=stack[top--];
                id[v]=newid;
            }while(v!=r);
        }
    }
    void solve()
    {
        ff(i,1,n)
        if(pre[i]==-1)
        tarjan(i,-1);
    }
    void build_map()
    {
        int  hash[N]={};
        cur=newhead;
        ff(i,1,n)
        {
            ffl(j,i)
            {
                int v=edge[j].v;
                if(id[i]!=id[v]&&hash[id[i]]!=id[v])
                add(id[v],id[i]),hash[id[i]]=id[v];
            }
        }
    }
    void solve_contain()
    {
        cc(contain,-1);
        cur=contain;
        ff(i,1,n)
        add(id[i],i);
    }
    bool two_sat()
    {
        solve();
        for(int i=1;i<=n;i+=2)
        if(id[i]==id[i+1])
        return false;
        return true;
    }
    void dfs(int r)
    {
        pre[r]=++Index;
        for(int i=newhead[r];i!=-1;i=edge[i].next)
         if(pre[edge[i].v]==-1) dfs(edge[i].v);
        low[++newid]=r;
    }
    void color_dfs(int r)
    {
        pre[r]=2;
        for(int i=newhead[r];i!=-1;i=edge[i].next)
        if(pre[edge[i].v]==-1) color_dfs(edge[i].v);
    }
    void Color()
    {
        cc(pre,-1);
       for(int i=nn;i>0;--i)
       if(pre[low[i]]==-1)
       {
          int a=edge[contain[low[i]]].v;
          int b;
          b=((a+1)/2)*4-1-a;
           pre[low[i]]=1;
           if(pre[id[b]]==-1)
           color_dfs(id[b]);
       }
    }
    void topo_sort()
    {
        nn=newid;
        Index=0;
        newid=0;
        cc(pre,-1);
        cc(low,-1);
        ff(i,1,nn)
        if(pre[i]==-1)
        dfs(i);
    }
    void taganswer()
    {
        cc(low,-1);
        ff(i,1,nn)
       if(pre[i]==1) {
         for(int j=contain[i];j!=-1;j=edge[j].next)
            low[edge[j].v]=1;
        }
    }
    void print_answer()
    {
        if(!two_sat())
        {printf("bad luck\n");
        return ;
        }
        solve_contain();
        build_map();
        topo_sort();
       Color();
       taganswer();
       bool xx=false;
        ff(i,3,n)
      if(low[i]==1)  {
            if(!xx)
            {
                printf("%d%c",(i+1)/2-1,i%2==0?'h':'w');
                xx=1;
            }
            else
            printf(" %d%c",(i+1)/2-1,i%2==0?'h':'w');
        }
        puts("");
    }
};
tarjan_seg G;
int b[500][500],na;
inline void addor(int i,int j,int val)
{
  if(val==0)
  {
      G.add(i*2,j*2);
      G.add(j*2,i*2);
      G.add(i*2-1,2*j);
      G.add(j*2-1,2*i);
      G.add(i*2-1,2*j-1);
      G.add(2*j-1,2*i-1);
  }
  else
  {
      G.add(i*2,j*2-1);
      G.add(j*2,i*2-1);
  }
}
inline void addxor(int i,int j,int val)
{
    if(val==0)
    {
        G.add(2*i,2*j);
        G.add(2*i-1,2*j-1);
        G.add(2*j,2*i);
        G.add(2*j-1,2*i-1);
    }
    else
    {
        G.add(2*i,2*j-1);
        G.add(2*j,2*i-1);
        G.add(2*i-1,2*j);
        G.add(2*j-1,2*i);
    }
}
inline void addand(int i,int j,int val)
{
    if(val==0)
    {
        G.add(2*i-1,2*j);
        G.add(2*j-1,2*i);
    }
    else
    {
        G.add(2*i,2*j-1);
        G.add(2*j,2*i-1);
        G.add(2*i-1,2*j-1);
        G.add(2*j-1,2*i-1);
        G.add(2*i,2*j);
        G.add(2*j,2*i);
    }
}
int main()
{
  while(~scanf("%d",&na))
{
    G.n=2*na;
    ff(i,0,na-1)
    ff(j,0,na-1)
    scanf("%d",&b[i][j]);
    bool f=true;
   ff(i,0,na-1)
   if(b[i][i]!=0) {f=false; break;}
   for(int i=0;f&&i<na;++i)
   ff(j,0,na-1)
   if(b[i][j]!=b[j][i])
   {
       f=false;
       break;
   }
   if(!f){printf("NO\n");continue;}
for(int k=0;f&&k<32;++k){
    G.init();
   ff(i,0,na-1)
{
  ff(j,i+1,na-1)
{
       int u;
       if(b[i][j]&(1<<k)) u=1;
       else  u=0;
       if(i%2==1&&j%2==1)
       {
           addor(i+1,j+1,u);
       }
       else
       if(i%2==0&&j%2==0)
       {
           addand(i+1,j+1,u);
       }
       else
       {
          addxor(i+1,j+1,u);
       }
}
}
f=G.two_sat();
}
printf("%s\n",f?"YES":"NO");
}
    return 0;
}