并查集题目

POJ 1611 The Suspects

题意是n个人，m组数，每组数的第一个数k表示这组数有k个数，求所有与0直接或间接有关系的数。

建立它们之间的关系后，只要查询下最终的根是0的个数就行，当然两个数中只能最小的数当根。

代码如下：

#include <iostream>
#include <cstdio>
using namespace std;
const int maxn = 3e4+10;
int fa[maxn], rank[maxn];
int n, m;
int find(int x){
    if(x == fa[x])return x;
    else return fa[x] = find(fa[x]);
}
void uni(int x, int y){
    x = find(x);
    y = find(y);
    if(x < y){
        fa[y] = x;
    }else{
        fa[x] = y;
    }
}
int main(){
    while(~scanf("%d",&n)&&n){
        scanf("%d",&m);
        int ans = 0;
        for(int i = 1; i < n; i ++){
            fa[i] = i;
            rank[i] = 0;
        }
        for(int i = 1; i <= m; i ++){
            int a, b, c;
            scanf("%d%d",&a,&b);
            for(int i = 1; i < a; i ++){
                scanf("%d",&c);
                uni(b, c);
            }
        }
        for(int i = 0; i < n; i ++){
            if(find(i) == 0) ans ++;
        }
        printf("%d
",ans);
    }
    return 0;
}

HDU 1314 How Many Tables

题意是n个人中，给定一些人之间的关系，直接或间接有关系的一组，求最终有多少组。

并查集，求最终的根是本身的有多少。

#include <iostream>
#include <cstdio>
using namespace std;
const int maxn = 1e3+10;
int fa[maxn];
int find(int x){
    if(fa[x] == x)return x;
    else return fa[x] = find(fa[x]);
}
void uni(int x, int y){
    x = find(x);
    y = find(y);
    if(x < y){
        fa[y] = x;
    }else{
        fa[x] = y;
    }
}
int main(){
    int t, n, m;
    scanf("%d",&t);
    while(t--){
        int ans = 0;
        scanf("%d%d",&n,&m);
        for(int i = 1; i <= n; i ++) fa[i] = i;
        for(int i  = 0; i < m; i ++){
            int a, b;
            scanf("%d%d",&a,&b);
            uni(a, b);
        }
        for(int i = 1; i <= n; i ++){
            if(fa[i] == i) ans ++;
        }
        printf("%d
",ans);
    }
    return 0;
}