hrbust1073 病毒 (并查集)

本文出自:http://blog.csdn.net/svitter


题意:不必赘述了。。一开始觉得直接遍历fa[i] == 0计数即可,因为本身就写了如果合并并查集,那么父节点就是0,但是没有考虑这种情况:fa[1] = 2, fa[2] = 2,  join(0, 1)此时fa[1]还是为2的,但是遍历fa[i]的时候不会将其计入。如是。


#include <iostream>
#include <stdio.h>
#include <string.h>
using namespace std;

int fa[50010];
int num;
int n, m;
void init()
{
    int i;
    for(i = 0 ; i < n; i++)
    {
        fa[i] = i;
    }
}

int find(int v)
{
    if(fa[v] != v)
        return fa[v] = find(fa[v]);
    return v;
}

void join(int x, int y)
{
    int fx = find(x), fy = find(y);
    if(fx != fy)
    {
        if(fx == 0)
            fa[fy] = 0;
        else if(fy == 0)
            fa[fx] = 0;
        else
            fa[fx] = fy;
    }
}

void ace()
{
    int i;
    int u, v;
    while(~scanf("%d%d", &n, &m))
    {
        num = 0;
        init();
        for(i = 0; i < m; i++)
        {
            scanf("%d%d", &u, &v);
            if(find(u) != find(v))
            {
                join(u, v);
            }
        }

        for(i = 0; i < n; i++)
        {
            if(find(i) == 0)
                num++;
        }
        printf("%d\n", num);
    }
}


int main()
{
    ace();
    return 0;
}


你可能感兴趣的:(C++,c,ACM,并查集)