Hdu1829 並查集+廣度優先搜索

....WA了我N次........

題目地址:    http://acm.hdu.edu.cn/showproblem.php?pid=1829

題目大意是這樣:給你多組"bug"和"bug"之間的關係(我理解爲交配關係...),讓你查看這些"bug"裏面有沒有同性戀.

好吧還是用測試數據來解釋吧.

2

3 3

1 2

2 3

1 3

4 2

1 2

3 4

先看紅色的測試數據:有三個"bug"其中1---2    2----3   1---3 有關係,但是這裏有一個矛盾就是

假如1爲雄性則2爲雌性3爲雄性,但是第三組數據可以看出1和3之間是"同性戀"...

而第二組測試數據則完全正常

1----2    3----4   符合要求.

題意理解了就好了.不過一開始想用深搜但是超時,換成廣搜就好多了.

首先將有關係的:"bug"的編號利用並查集處理這樣在廣搜的時候可以將"獨立"的元素送入隊列中,我數字1代表雄性 2代表雌性

"獨立"的元素看成雄性.

還是以第三組數據爲例: 在利用並查集處理之後.

group[1] = 2;  group[2] = 3;group[3] = 3;

這樣廣搜的時候  3 最先送入隊列.

根據事先建立的關係圖. 

3------1           3爲雄性   1爲雌性

1------2           1爲雌性    2爲雄性

2------3           2爲雄性    3爲雄性!!!!!好矛盾出現了.

所以這時候程序返回值說明這一組"bug"裏面有同性戀.

#include <iostream>
#include <queue>
#include <algorithm>
#include <cstring>
using namespace std;

int x,y;
bool maps[2001][2001];
int sex[2001];
int group[2001];

void init()
{
	int i;
	for(i=1;i<=x;i++)
		group[i]=i;
}

int Find(int z)
{
	int i,f=z;
	while(f!=group[f])
		f=group[f];
        
        //這一段老實說是網上看到的,因爲沒有這一段運行時間會長一些
        while(z!=f)
	{
		i=group[z];
		group[z]=f;
		z=i;
	}
	return f;
}
int bfs()
{
	int i,t;
	queue<int>q;
	for(i=1;i<=x;i++)
	{
		if(i==group[i])
		{
			sex[i]=1;
			q.push(i);
		}
	}
	while(!q.empty())
	{
		t=q.front();
		q.pop();
		for(i=1;i<=x;i++)
		{
			if(maps[t][i])
			{
				if(sex[i]==0)
				{
					if(sex[t]==1)sex[i]=2;
					else if(sex[t]==2)sex[i]=1;
					q.push(i);
				}
				else
				{
					if(sex[t] == sex[i])return 1;
				}

			}
			
		}
	}
	return 0;
}
int main()
{
	int i,n;
	int cx,cy,count;
	scanf("%d",&n);
	count=1;
	for(count=1;count<=n;count++)
	{
		memset(sex,0,sizeof(sex));
		memset(maps,0,sizeof(maps));
		scanf("%d%d",&x,&y);
		init();
		for(i=0;i<y;i++)
		{
			scanf("%d%d",&cx,&cy);
			maps[cx][cy]=maps[cy][cx]=1;
			if(Find(cx)!=Find(cy))
			{
				group[Find(cx)] = Find(cy);
			}
		}
		if(bfs())
		{
			printf("Scenario #%d:\n",count);
			printf("Suspicious bugs found!\n\n");
		}
		else
		{
			printf("Scenario #%d:\n",count);
			printf("No suspicious bugs found!\n\n");
		}
	}
	return 0;
}