The Suspects POJ

The Suspects

题目：编号为0的人有传染病，同组中只要有一个人有传染病，该组的人都被看做有传染病，一个人可以在多组中，问有多少人有传染病。

思路：并查集，需要压缩并查集的树，编号小的点优先作为祖先（0为root），并查集过程中传递祖先的同时传递祖先是否是病人，最后再次遍历所有人，使得祖先是病人子孙不是病人的情况解决，因为我们压缩了并查集的树，所以这个过程的复杂度不高。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <vector>
#include <cmath>

using namespace std;

#define ll long long
#define pb push_back
#define fi first
#define se second

const int N = 3e4 + 10;
bool vis[N];
int n, m;

struct node{
    int root, vis;
}fa[N];

pair<int ,int > Find(int now){
    if(fa[now].root == now){
        //传递祖先和病人信息
        return make_pair(fa[now].root, fa[now].vis);
    }else{
        pair<int ,int > tmp = Find(fa[now].root);
        fa[now].root = tmp.fi;
        fa[now].vis = tmp.se;
        return tmp;
    }
}

void Union(int x, int y)
{
    int fax = Find(x).fi;
    int fay = Find(y).fi;
    //编号小的人作为祖先
    if(fax > fay) swap(fax, fay);
    fa[fay].root = fax;
}


void solve()
{
    while(~scanf("%d%d", &n, &m) && n + m){
        for(int x = 0; x < n; ++x){
            fa[x].root = x;
            fa[x].vis = 0;
        }
        fa[0].vis = 1;
        int num, idx, idy;
        for(int x = 0; x < m; ++x){
            scanf("%d%d", &num, &idx);

            for(int y = 1; y < num; ++y){
                scanf("%d", &idy);
                Union(idx, idy);
            }
        }

        //再次传递病人信息
        for(int i = 0; i < n; ++i){
            Find(i);
        }

        int suspects = 0;
        for(int x = 0; x < n; ++x){
            if(fa[x].vis) ++suspects;
        }

        //printf("suspects = %d
", suspects);
        printf("%d
", suspects);
    }
}

int main()
{

    solve();

    return 0;
}