HDU 1213 How Many Tables (并查集，常规)

 1 #include <bits/stdc++.h>
 2 #define LL long long
 3 using namespace std;
 4 const int N=1005;
 5 int pre[N]; //结点i的上级
 6 inline void init(int n){for(int i=1; i<=n; i++) pre[i]=i;}  //将上级初始化为自己
 7 
 8 int find(int x) //找x的上级
 9 {
10     int k=x;
11     while(pre[x]!=x)    x=pre[x];   //找上级
12     int tmp;
13     while(pre[k]!=x) //优化:路径压缩
14     {
15         tmp=pre[k];
16         pre[k]=x;
17         k=tmp;
18     }
19     return x;
20 }
21 
22 void joint(int a,int b)  //将a和b合并为一个集合
23 {
24     a=find(a);
25     b=find(b);
26     if(a!=b)    pre[a]=b;
27 }
28 
29 int check(int n)    //查查到底需要几张桌子
30 {
31     set<int> sett;
32     for(int i=1; i<=n; i++)    find(i); //防漏网之鱼，将所有人的老大推到最顶。
33     for(int i=1; i<=n; i++)    sett.insert(pre[i]); 
34     return sett.size();
35 }
36 int main()
37 {
38     //freopen("e://input.txt", "r", stdin);
39     int t, n, m, a, b;
40     cin>>t;
41     while(t--)
42     {
43         scanf("%d%d", &n, &m);
44         init(n);
45         for(int i=0; i<m; i++)
46         {
47             scanf("%d%d",&a,&b);
48             joint(a,b);
49         }
50         printf("%d\n",check(n));
51     }
52     return 0;
53 }