首页 > 代码库 > HDU1824

HDU1824

2-SAT入门题,将两个队员看成一个点即可,即用一条边将两个队员连起来,然后就是2-SAT裸题。

#include <stdio.h>
#include <string.h>
#include <string>
#include <iostream>
#include <algorithm>
#include <vector>
#include <math.h>
#include <map>
#include <queue>
#include <stack>
#include <set>
#define M 3005
#define LL long long
#define Ld __int64
#define eps 0.000001
#define INF 999999999
#define MOD 112233
#define MAX 26
using namespace std;

vector<int> G[M];  
//dfn数组表示dfs时到达i点的时间,indx表示时间
vector<int> P[M];
int dfn[M],low[M],sccno[M],scc_cnt;  
int indx;  
int num[M];
stack<int> s;  

void Tarjan(int u)  
{  
	indx++;
	dfn[u]=low[u]=indx;   //为结点u设定次序编号和low初值
	s.push(u);            //将结点u压入栈中
	for(int i=0;i<G[u].size();i++)   //枚举每条边
	{   
		int v=G[u][i];  
		if(!dfn[v])			//若结点v未被访问过
		{  
			Tarjan(v);      //继续往下找
			low[u]=min(low[u],low[v]);   
		}  
		else if(!sccno[v])  //若结点v在栈中
		{  
			low[u]=min(low[u],dfn[v]);  
		}  
	}  
	if(low[u]==dfn[u])    //如果结点u是强连通分量的根
	{  
		scc_cnt++;  
		for(;;)  
		{  
			int x=s.top();  //讲v退栈,为该强连通分量中一个顶点
			s.pop();  
			sccno[x]=scc_cnt;  
			num[scc_cnt]++;  //记录每个强连通分量的结点数
			if(x==u)  
				break;  
		}  
	}  
}  

void find_scc(int n)  
{  
	indx=scc_cnt=0;  
	memset(num,0,sizeof(num));
	memset(sccno,0,sizeof(sccno));  
	memset(dfn,0,sizeof(dfn));  
	for(int i=0;i<n*3;i++)  
		if(!dfn[i])  
			Tarjan(i);  
} 

int C[M],S[M];			//C存的是队长,S存的是队员

int main()
{
	int n,m;
	while(~scanf("%d%d",&n,&m))
	{
		memset(C,0,sizeof(C));
		memset(S,0,sizeof(S));
		for(int i=0;i<n*3;i++)
		{
			G[i].clear();
			P[i].clear();
		}
		for(int i=0;i<n;i++)
		{
			int a;
			scanf("%d%d%d",&C[i],&S[i],&a);
			G[S[i]].push_back(a);			//两个队员之间建边
			G[a].push_back(S[i]);
			P[C[i]].push_back(S[i]);		//队长C[i]的另外两个队员是谁
			P[C[i]].push_back(a);
			P[S[i]].push_back(C[i]);		//这个队员的队长是谁
			P[a].push_back(C[i]);
		}
		while(m--)
		{
			int u,v;
			scanf("%d%d",&u,&v);
			int k=P[u].size(),o=P[v].size();
			G[u].push_back(P[v][0]);
			G[v].push_back(P[u][0]);
		}
		find_scc(n);
		int flag=0;
		for(int i=0;i<n;i++)
		{
			if(sccno[C[i]]==sccno[S[i]])
			{
				flag=1;
				break;
			}
		}
		if(flag==1)
			printf("no\n");
		else
			printf("yes\n");
	}
	return 0;
}