hdu 4115 2-sat

来源:互联网 发布:sm3算法 编辑:程序博客网 时间:2024/06/04 17:54

又是神奇的建图题,这道题充分体现了2-sat选了谁必须选谁的性质

某一轮如果选了必输的方法,则要连一条边到对立点;即如果选a必输,则连一条边a->a‘

总之,要不断找矛盾

贴贴很挫的代码

View Code
#include<stdio.h>
#include<string.h>
#include<vector>
#include<algorithm>
using namespace std;
const int MAX = 20010;
vector<int> edge[MAX];
int st[MAX];
int dfn[MAX],low[MAX];
int top,btype,tdfn;
int belong[MAX];
bool ins[MAX];
void dfs(int s)
{
int i,t;
dfn[s]=low[s]=++tdfn;
ins[s]=true;
st[++top]=s;
for(i=0;i<edge[s].size();i++)
{
t=edge[s][i];
if(!dfn[t])
{
dfs(t);
if(low[t]<low[s]) low[s]=low[t];
}
else if(ins[t] && dfn[t]<low[s]) low[s]=dfn[t];
}
if(dfn[s]==low[s])
{
btype++;
do
{
t=st[top--];
ins[t]=false;
belong[t]=btype;
}while(t!=s);
}
}
void SCC(int n)
{
int i;
top=btype=tdfn=0;
memset(ins,false,sizeof(ins));
memset(dfn,0,sizeof(dfn));
for(i=1;i<=n;i++)
if(!dfn[i])
dfs(i);
}
int bob;
int alice[MAX];
int main()
{
int i,j,k,t;
int n,m;
int a,b,cases=1;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&m);
for(i=0;i<=2*n;i++)
edge[i].clear();
for(i=1;i<=n;i++)
{
scanf("%d",&bob);
if(bob==1)
{
alice[i]=1;
alice[i+n]=2;
}
if(bob==2)
{
alice[i]=2;
alice[i+n]=3;
}
if(bob==3)
{
alice[i]=3;
alice[i+n]=1;
}
}
for(i=0;i<m;i++)
{
scanf("%d%d%d",&a,&b,&k);
if(k==1)
{
if(alice[a]==alice[b])
{
edge[a].push_back(b+n);
edge[b].push_back(a+n);
}
if(alice[a]==alice[b+n])
{
edge[a].push_back(b);
edge[b+n].push_back(a+n);
}
if(alice[a+n]==alice[b])
{
edge[a+n].push_back(b+n);
edge[b].push_back(a);
}
if(alice[a+n]==alice[b+n])
{
edge[a+n].push_back(b);
edge[b+n].push_back(a);
}
}
if(k==0)
{
if(alice[a]!=alice[b])
{
edge[a].push_back(b+n);
edge[b].push_back(a+n);
}
if(alice[a]!=alice[b+n])
{
edge[a].push_back(b);
edge[b+n].push_back(a+n);
}
if(alice[a+n]!=alice[b])
{
edge[a+n].push_back(b+n);
edge[b].push_back(a);
}
if(alice[a+n]!=alice[b+n])
{
edge[a+n].push_back(b);
edge[b+n].push_back(a);
}
}
}
SCC(2*n);
int flag=1;
for(i=1;i<=n;i++)
{
if(belong[i]==belong[i+n])
{
flag=0;
break;
}
}
if(flag) printf("Case #%d: yes\n",cases++);
else printf("Case #%d: no\n",cases++);
}
return 0;
}



原创粉丝点击