LightOj 1251 Forming the Council(2-sat)

原题地址：点击打开链接

#include<stdio.h>
#include<vector>
#include<queue>
#include<string.h>
#include<stack>
using namespace std;
int m;
int time;
int low[20000];
int dfn[20000];
int cmp[20000];
int in[20000];
int color[20000];
int fp[20000];
bool instack[20000];
int count;
int k;
vector<int>edge[20000];
vector<int>g[20000];
void add_edge(int u,int v)
{
int x=u,y=v;
if(u<0) x=-u+m;
if(v<0) y=-v+m;
if(x>m) edge[x-m].push_back(y);
else edge[x+m].push_back(y);
if(y>m) edge[y-m].push_back(x);
else edge[y+m].push_back(x);
}
stack<int>S;
void tarjan(int u)
{
dfn[u]=low[u]=++time;
S.push(u);
instack[u]=true;
for(int i=0;i<edge[u].size();i++)
{
int v=edge[u][i];
if(dfn[v]==-1)
{
tarjan(v);
low[u]=min(low[u],low[v]);
}
else if(instack[v])
{
low[u]=min(dfn[v],low[u]);
}
}
if(dfn[u]==low[u])
{
count++;
int a;
do
{
a=S.top();
S.pop();
cmp[a]=count;
instack[a]=false;
}while(a!=u);
}
}
void suodian() //建新图
{
memset(in,0,sizeof(in));
for(int i=1;i<=count;i++)
g[i].clear();
for(int i=1;i<=2*m;i++)
{
for(int j=0;j<edge[i].size();j++)
{
int u=edge[i][j];
if(cmp[i]!=cmp[u]) //反向建图
{
g[cmp[u]].push_back(cmp[i]);
in[cmp[i]]++;
}

}
}
}
void tpsort()
{
memset(color,-1,sizeof(color));
int i,u;
queue<int>que;
for(i=1;i<=count;i++)
{
if(in[i]==0)
que.push(i); //将入度为0的顶点加入队列
}
while(!que.empty())
{
u=que.front();
que.pop() ;
if(color[u]==-1)
{
color[u]=1;
color[fp[u]]=0;
}
for(i=0;i<g[u].size();i++)
{
int v=g[u][i];
if(--in[v]==0)
que.push(v);
}
}
}
void solve()
{
int res=0,i;
count=0;
time=0;
memset(dfn,-1,sizeof(dfn));
memset(cmp,-1,sizeof(cmp));
memset(instack,false,sizeof(instack));
for(i=1;i<=2*m;i++)
{
if(dfn[i]==-1)
tarjan(i);
}
printf("Case %d: ",++k);
for(i=1;i<=m;i++)
{
if(cmp[i]==cmp[i+m])
{
printf("No\n");
return ;
}
else
{
fp[cmp[i]]=cmp[i+m];
fp[cmp[i+m]]=cmp[i];
}
}
suodian();
tpsort();
for(i=1;i<=m;i++)
{
if(color[cmp[i]]==1)
res++;
}
printf("Yes\n%d",res);
for(i=1;i<=m;i++)
{
if(color[cmp[i]]==1)
printf(" %d",i);
}
printf("\n");
}
int main()
{
int i,t,n,u,v;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&m);
for(i=1;i<=2*m;i++)
edge[i].clear();
for(i=0;i<n;i++)
{
scanf("%d%d",&u,&v);
add_edge(u,v);
}
solve();
}
return 0;
}