POJ1611 The Suspects(并查集)

来源:互联网 发布:苹果电脑安装软件 编辑:程序博客网 时间:2024/05/01 13:33

第一次学并查集,总算对它有了一定的了解,基本就是连通的集合的问题

参考博客:基本概念   结合一些具体问题

题意:

学校中一共有m个学生,n个组,学生编号从0~m-1,现在学生0感冒了,如果组中有一人感冒,剩下的人都会感冒,给出每个组的具体学生序号,求出有多少人会感冒

要点:

并查集裸题,要点是先将所有组合并,最后遍历找于0同一根节点的数量即为感冒人数,要注意0不一定是最终的根节点


15310408Seasonal1611Accepted404K16MSC++915B2016-03-24 22:21:30

#include<stdio.h>#include<string.h>#include<stdlib.h>#define maxn 40000int p[maxn], rank[maxn];int m, n;void init(){for (int i = 0; i < m; i++){p[i] = i;rank[i] = 0;//记录树的深度}}int find(int x)//找到根节点{if (x == p[x]) return x;return p[x] = find(p[x]);//状态压缩,使所有节点都指向根节点}void merge(int x,int y)//合并集合{x = find(x);y = find(y);if (x == y) return;if (rank[x] > rank[y])//将深度小的树指向深度大的树p[y] = x;else{p[x] = y;if (rank[x] == rank[y])//深度如果相同,深度大的树深度要+1rank[y]++;}}int main(){while (~scanf("%d%d", &m, &n), n + m){init();while (n--){int t,x,y;scanf("%d", &t);scanf("%d", &x);//每组的第一个数for (int i = 1; i < t; i++){scanf("%d", &y);merge(x, y);//将每组的后续数都与第一个数合并}}int sum = 0;int xx = find(0);//注意0不一定是根节点,可能只是在树中for (int i = 0; i < m; i++)if (find(i) == xx)sum++;printf("%d\n", sum);}return 0;}


0 0
原创粉丝点击