（判断二分图）hud 5971&2016ICPC大连 A - Wrestling Match

最新推荐文章于 2021-01-29 18:54:24 发布

假装有字

最新推荐文章于 2021-01-29 18:54:24 发布

阅读量190

点赞数

文章标签：二分图染色（非联通块）

本文链接：https://blog.csdn.net/cjj1226/article/details/103081695

版权

本文介绍了一种基于并查集和深度优先搜索(DFS)的二分图染色算法，用于解决WrestlingMatch问题。通过记录节点的边信息和颜色状态，实现对图中每个节点的有效染色，确保相邻节点颜色不同，从而判断图是否为二分图。

摘要由CSDN通过智能技术生成

Wrestling Match

思路：这题就是判断是否是二分图（非联通块）也就是二分图染色，用了并查集加深搜。

#include <bits/stdc++.h>
using namespace std;
#define ll long long
int n,m,x,y,last[20005],num,f[1005],a,b,co[1005],to[20005],flag,z;
//last记录前一个相同节点的边，f[i]记录i的最后一条边对应的编号 
//to记录边所连接的另一个节点 ，flag判断是否能构成二分图 
void add(int a,int b)//并查集 
{
	last[++num]=f[a];
	to[num]=b;
	f[a]=num;
}
void change(int s,int color)//染色，深搜 
{
	co[s]=color;
	s=f[s];
	while(s!=0)
	{
		if(co[to[s]]==color)
		{
			flag=1;
			break;
		}
		if(co[to[s]]==0)
			change(to[s],-color);
		s=last[s];
	}
}
int main()
{
	while(~scanf("%d %d %d %d",&n,&m,&x,&y))
	{
		num=0,flag=0;
		for(int i=1;i<=n;i++)
			f[i]=0,co[i]=0;
		for(int i=1;i<=m;i++)
		{
			last[i*2-1]=0,last[i*2]=0;
			scanf("%d %d",&a,&b);
			add(a,b);
			add(b,a);
		}
		for(int i=0;i<x;i++)//先从确定颜色的点开始染色 
		{
			scanf("%d",&z);
			if(co[z]==-1)
				flag=1;
			if(flag==0)
				change(z,1);
		}
		for(int i=0;i<y;i++)
		{
			scanf("%d",&z);
			if(co[z]==1)
				flag=1;
			if(flag==0)
				change(z,-1);
		}
		for(int i=1;i<=n;i++)//遍历没有染色却还有边的点进行染色 
		{
			if(flag)
				break;
			if(f[i]!=0&&co[i]==0)
				change(i,1);
			if(co[i]==0)	//如果存在孤立的点且不知其颜色 
			{
				//cout<<i<<endl;
				flag=1;
			}
		}
		if(flag)
			printf("NO\n");
		else
			printf("YES\n");
	}
	return 0;
}