hdu 4115 石头剪子布（2-sat问题）

/*
意甲冠军：石头剪子布，目前已知n周围bob会有什么，对alice限制。供u,v,w；设w=0说明a,b回合必须出的一样
否则，必须不一样。alice假设输一回合就输了，否则就赢了
解：
2-sat
alice有两个选择要么平手要么赢。
对于第u回合，alice能够出au，bu；
对于第v回合，alice能够出av，bv；
当w=0那么第u回合和第v回合必须同样
比較au和bu。bv是否矛盾，假设矛盾建两条边
比較av和bu。

bv是否矛盾，假设矛盾建两条边
当w=1第u回合和第v回合必须不同样
比較au和bu。bv是否矛盾，假设矛盾建两条边
比較av和bu。bv是否矛盾，假设矛盾建两条边
*/
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
using namespace std;
#define N 21000
#define NN 11000
struct node
{
int u,v,next;
} bian[NN*20];
int head
,yong,low
,dfn
,belong
,ans,top,index,stac
,vis
;
void init()
{
memset(head,-1,sizeof(head));
yong=index=ans=top=0;
memset(vis,0,sizeof(vis));
memset(dfn,0,sizeof(dfn));
}
void addedge(int u,int v)
{
bian[yong].v=v;
bian[yong].next=head[u];
head[u]=yong++;
}
void tarjan(int u)
{
low[u]=dfn[u]=++index;
stac[++top]=u;
vis[u]=1;
int i;
for(i=head[u]; i!=-1; i=bian[i].next)
{
int v=bian[i].v;
if(!dfn[v])
{
tarjan(v);
low[u]=min(low[u],low[v]);
}
else if(vis[v])
low[u]=min(low[u],dfn[v]);
}
if(low[u]==dfn[u])
{
ans++;
int t;
do
{
t=stac[top--];
belong[t]=ans;
vis[t]=0;
}
while(t!=u);
}
}
int slove(int n)
{
int i;
for(i=0; i<n*2; i++)
if(!dfn[i])
tarjan(i);
// printf("%d\n",ans);
for(i=0; i<n; i++)
if(belong[i]==belong[i+n])
return 0;
return 1;
}
void Switch(int f,int &au,int &av)
{
au=f;
av=(f+1)%3;
return ;
}
int a
;
int main()
{
int t,n,m,i,k=0,u,v,w;
int au,av,bu,bv;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&m);
for(i=0; i<n; i++)
{
scanf("%d",&a[i]);
a[i]--;
}
init();
for(i=0; i<m; i++)
{
scanf("%d%d%d",&u,&v,&w);
u--;
v--;
Switch(a[u],au,av);
Switch(a[v],bu,bv);
if(!w)
{
if(au!=bu)
{
addedge(u,v+n);
addedge(v,u+n);
}
if(au!=bv)
{
addedge(u,v);
addedge(v+n,u+n);
}
if(av!=bu)
{
addedge(u+n,v+n);
addedge(v,u);
}
if(av!=bv)
{
addedge(u+n,v);
addedge(v+n,u);
}
}
else
{
if(au==bu)
{
addedge(u,v+n);
addedge(v,u+n);
}
if(au==bv)
{
addedge(u,v);
addedge(v+n,u+n);
}
if(av==bu)
{
addedge(u+n,v+n);
addedge(v,u);
}
if(av==bv)
{
addedge(u+n,v);
addedge(v+n,u);
}
}
}
if(!slove(n))
printf("Case #%d: no\n",++k);
else
printf("Case #%d: yes\n",++k);
}
return 0;
}