poj2492_bfs_并查集

题意：

给的就是n个虫子 m对虫子发生过关系 然后问有没有虫子是同性恋。

分析：题目等价为，n个点，m条边,能否仅用两种颜色染完所有点，并使每条边的两个点不同色。

bfs,遍历每个顶点，如果相邻顶点染有相同的颜色，说明有同性恋！

代码：

#include <iostream>
#include <stdio.h>
#include <memory.h>
#include <queue>
using namespace std;
const int maxnum=2001;
bool array[maxnum][maxnum];
bool use[maxnum];
bool color[maxnum];
int m,n;
bool flag;
queue<int> q;

void Init()
{
    int i;
    int a,b;
    scanf("%d%d",&m,&n);
    memset(array,false,sizeof(array));
    memset(use,false,sizeof(use));
    for(i=1;i<=n;i++)
    {
        scanf("%d%d",&a,&b);
        array[a][b]=true;
        array[b][a]=true;
    }
    flag=true;

    while(!q.empty())
        q.pop();
}

void bfs(int u)
{
    int v,i;
    use[u]=true;
    color[u]=true;
    q.push(u);
    while(!q.empty())
    {
        v=q.front();
        q.pop();
        for(i=1;i<=m;i++)
            if(i!=v && array[v][i])
            {
                if(!use[i])
                {
                    color[i]=!color[v];
                    use[i]=true;
                    q.push(i);
                }
                else
                {
                    if(color[i]==color[v])
                    {
                        flag=false;
                        return ;
                    }
                }
            }
    }
}

int main()
{
    int num;
    scanf("%d",&num);
    int i,k;
    for(k=1;k<=num;k++)
    {
        Init();
        for(i=1;i<=m;i++)
            if(!use[i] && flag)
                bfs(i);

        printf("Scenario #%d:\n",k);
        if(!flag)
            printf("Suspicious bugs found!\n\n");
        else
            printf("No suspicious bugs found!\n\n");
    }
    return 0;
}

网上有用并查集的，小小的学习了一下。内存比bfs小的多。

并查集思想：

1.用father[i]表示i的根节点，sex[i]表示与它的根节点相比，true表示性别相同，false表示性别不同

2.对于每对a,b。查找father[a]和father[b],不同进行组合，相同进行判断。

3.在a,b的根节点不同的情况下，需要更新sex[father[a]]和sex[fathe[b]]。

首先求出a,b相对于根节点的sex[a]和sex[b].举例

输入1  2后有。father[2]=1,sex[2]=false;

输入2 3 后：首先得出father[2]=1,father[3]=3,不同。

更新father:father[3]=1

更新sex:

因为根节点1和3相对自己是同性，而2相对于1是异性，所以,这样sex[2]^sex[3]=不同两项异或，为1，

所以sex[3]=true;这样3就与1同性。

代码：

#include <iostream>
#include <stdio.h>
using namespace std;
const int maxnum=2001;
int root[maxnum];
int sex[maxnum];//相对于父节点的性别,true表示是同性，false表示异性
int m,n;
bool flag;

void Init()
{
    scanf("%d%d",&m,&n);
    int i;
    for(i=1;i<=m;i++)
    {
        root[i]=i;
        sex[i]=true;
    }
    flag=true;
}

int find(int i)
{
    int t=i;
    bool f=true;
    while(t!=root[t])
    {
        if(!sex[t])  //sex[t]==false;
            f=!f;
        t=root[t];
    }
    root[i]=t;  //状态压缩
    sex[i]=f;   //当前节点相对于根节点的性别
    return root[i];
}

void function()
{
    int i;
    int a,b;
    int ra,rb;
    for(i=1;i<=n;i++)
    {
        scanf("%d%d",&a,&b);
        if(flag)
        {
            ra=find(a);
            rb=find(b);
            if(ra==rb)
            {
                if(sex[a]==sex[b])   //找到同性恋
                    flag=false;
            }
            else
            {
                root[rb]=ra;   //合并两个集合
                sex[rb]=sex[a]^sex[b]; //更新sex[rb]
            }
        }
    }
}

int main()
{
    int num;
    int k;
    scanf("%d",&num);
    for(k=1;k<=num;k++)
    {
        Init();
        function();
        printf("Scenario #%d:\n",k);
        if(!flag)
            printf("Suspicious bugs found!\n\n");
        else
            printf("No suspicious bugs found!\n\n");
    }
    return 0;
}

 

tjuoj 1706