首页 > 其他 > 详细

POJ1611-The Suspects-并查集

时间:2016-02-02 01:08:51      阅读:336      评论:0      收藏:0      [点我收藏+]

记录元素个数的并查集。

利用sz数组保存并查集的大小。每次union时,把小的集合并到大的中去,并更新sz数组。

#include <cstdio>
#include <algorithm>

using namespace std;

int n,m,k;

int fa[30010],sz[30010];

int Find(int x)
{
    while(x != fa[x])
    {
        fa[x] = fa[fa[x]];
        x = fa[x];
    }
    return x;
}

void Union(int p,int q)
{
    int fp = Find(p),fq = Find(q);
    if(fp == fq) return;
    if(sz[fp] < sz[fq])
    {
        fa[fp] = fq;
        sz[fq] += sz[fp];
    }
    else
    {
        fa[fq] = fp;
        sz[fp] += sz[fq];
    }
}

int main()
{
    while(scanf("%d%d",&n,&m) && (n||m))
    {
        for(int i=0;i<n;i++)
        {
            fa[i] = i;
            sz[i] = 1;
        }
        for(int i=0;i<m;i++)
        {
            scanf("%d",&k);
            int root = 0,p;
            for(int i=0;i<k;i++)
            {
                scanf("%d",&p);
                if(!i) root = p;
                else Union(p,root);
            }
        }
        printf("%d\n",sz[Find(0)]);
    }
}

 

POJ1611-The Suspects-并查集

原文:http://www.cnblogs.com/helica/p/5176588.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!