2016 大连 A HDU 5971 Wrestling Match · 模拟

用并查集先将所有的联通块处理出来，然后对于每个联通块，选定已经确定的好/坏选手为起点开始判断，直到遍历完整个联通块。

特殊情况：如果整个联通块中都没有已经确定的选手，那么整个联通块都是无法判断的

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <vector>
using namespace std;
#define f(i,x,y) for (int i=x;i<=y;i++)

const int N=1e3+10;
vector<int> a[N];
int n,m,data[N],x,y;
int fa[N],vis[N],check[N];
int u[10005],v[10005];

int find(int x){
	return fa[x]==x?x:fa[x]=find(fa[x]);
}

bool work(){
	int q[N],l,r;
	memset(vis,0,sizeof vis);	
	memset(check,0,sizeof check); //并查集 
	f(i,1,n) fa[i]=i;
	int fx,fy;
	f(i,1,m){
		fx=find(u[i]);fy=find(v[i]);
		if (fx!=fy) fa[fx]=fy;
	}
	
	f(i,1,n) if (!data[i] && !a[i].size()) return 0;
	
	r=0;
	f(i,1,n) if (data[i]){
		q[++r]=i;
		check[find(i)]=1;
		vis[i]=1;
	}
	f(i,1,n) if (!data[i] && !check[find(i)]){
		data[i]=1;
		q[++r]=i;
		check[find(i)]=1;
		vis[i]=1;
	}
	
	l=1;
	while (l<=r){
		int now=q[l];
		if (a[now].size())
		f(i,0,a[now].size()-1){
			int x=a[now][i];
			if (data[x] && data[x]+data[now]) return 0;
			data[x]=-data[now];
			if (!vis[x]) q[++r]=x;
			vis[x]=1;
		}
		++l;
	}
	return 1;
}

int main(){
	while (scanf("%d%d%d%d",&n,&m,&x,&y)!=EOF){
		;
		f(i,1,n) a[i].clear();
		memset(data,0,sizeof data);
		int p,q;
		f(i,1,m) {
			scanf("%d%d",&p,&q);
			u[i]=p;v[i]=q;
			a[p].push_back(q);
			a[q].push_back(p);
		}
		f(i,1,x) {
			scanf("%d",&p);data[p]=1;
		}
		f(i,1,y) {
			scanf("%d",&p);data[p]=-1;
		}
		if (work()) puts("YES");
			else puts("NO");
	}
	return 0;
}