当前位置: 代码迷 >> 综合 >> zoj - 1789 - The Suspects
  详细解决方案

zoj - 1789 - The Suspects

热度:16   发布时间:2024-01-10 14:00:38.0

题意:有n个学生,编号为0到n-1,m个群,给出每个群的学生,开始学生0被怀疑得了SARS,于是,人传人,被怀疑的学生所属群的所有学生都被怀疑,问共有几个学生被怀疑。

题目链接:http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=789

——>>简单的并查集题目,将每个群的后一个学生与前一个学生judge,统计结点数输出即可。

带路径压缩递归写法:

#include <iostream>using namespace std;const int maxn = 30000 + 10;        //0 < n <= 30000,开30000+10够矣int fa[maxn], height[maxn], cnt[maxn];      //fa[i]为i的父亲,height[i]为以i为根的树的高度,cnt[i]为以i为根的树的叶子的结点数int find(int x)     //找根
{return fa[x] == x ? x : (fa[x] = find(fa[x]));
}bool judge(int x, int y)        //判断、合并操作
{int new_x = find(x);        //找x的根int new_y = find(y);        //找y的根if(new_x == new_y) return 0;fa[new_y] = new_x;      //将树new_y合并到new_x中cnt[new_x] += cnt[new_y];       //更新树new_x的叶子数return 1;
}int main()
{int n, m, i, j, num, first, temp;       //num为一个GROUPR的人数,first为一个GROUP的第一个人while(cin>>n>>m){if(n == 0 && m == 0) return 0;for(i = 0; i < n; i++)      //初始化{fa[i] = i;      //建立n棵树,自己就是树根cnt[i] = 1;     //只有自己,结点数为1}for(i = 0; i < m; i++){cin>>num>>first;        //先输入第一个人,后面的人每一个都与前一个judgefor(j = 0; j < num-1; j++){cin>>temp;judge(first, temp);}}cout<<cnt[find(0)]<<endl;}return 0;
}

不那么地压缩写法:

#include <iostream>using namespace std;const int maxn = 30000 + 10;        //0 < n <= 30000,开30000+10够矣int fa[maxn], height[maxn], cnt[maxn];      //fa[i]为i的父亲,height[i]为以i为根的树的高度,cnt[i]为以i为根的树的叶子的结点数int find(int x)     //找根
{while(x != fa[x]){x = fa[x];}return x;
}bool judge(int x, int y)        //判断、合并操作
{int new_x = find(x);        //找x的根int new_y = find(y);        //找y的根if(new_x == new_y) return 0;else{if(height[new_x] > height[new_y])       //当树new_x的高度 > 树new_y的高度时{fa[new_y] = new_x;      //将树new_y合并到new_x中cnt[new_x] += cnt[new_y];       //更新树new_x的叶子数}else if(height[new_x] == height[new_y])     //当树new_x的高度 == 树new_y的高度时{fa[new_y] = new_x;      //将树new_y合并到new_x中cnt[new_x] += cnt[new_y];       //更新树new_x的叶子数height[new_x]++;        //树new_x的高度加1}else        //当树new_x的高度 == 树new_y的高度时{fa[new_x] = new_y;      //将树new_x合并到new_y中cnt[new_y] += cnt[new_x];       //更新树new_y的叶子数}return 1;}
}int main()
{int n, m, i, j, num, first, temp;       //num为一个GROUPR的人数,first为一个GROUP的第一个人while(cin>>n>>m){if(n == 0 && m == 0) return 0;for(i = 0; i < n; i++)      //初始化{fa[i] = i;      //建立n棵树,自己就是树根cnt[i] = 1;     //只有自己,结点数为1height[i] = 1;      //只有自己,为1层}for(i = 0; i < m; i++){cin>>num>>first;        //先输入第一个人,后面的人每一个都与前一个judgefor(j = 0; j < num-1; j++){cin>>temp;judge(first, temp);first = temp;       //更新前者}}cout<<cnt[find(0)]<<endl;}return 0;
}