POJ2524(Ubiquitous Religions)--并查集

题目链接：http://poj.org/problem?id=2524

学校有n个学生，但是你不能直接问学生的信仰，不然他会感到很不舒服的.一个方法是问m对同学，是否信仰同一宗教。根据这些数据,

计算学校最多有多少种宗教信仰的。

思路：使用并查集，一开始假设大家都各自信仰一个宗教，那么总的数目ans就是学生数目，

当发现有一对学生信仰同一个宗教，那么ans减1。

#include<iostream>
#include<stdio.h>
#include<cstring>
#include<cmath>
#include<vector>
#include<stack>
#include<map>
#include<set>
#include<list>
#include<queue>
#include<string>
#include<algorithm>
#include<iomanip>
using namespace std;


struct node
{
    /* data */
    int date;//结点对应的下标
    int parent;//双亲结点
    int rank;//结点对应秩（并查集树的深度）
    int relation; //与父节点的关系
    int total;
};

class DisJoinSet
{
protected:
    int n;//元素个数
    node *tree;//并查集元素数组
public:
    int ans;
    DisJoinSet(int n );
    ~DisJoinSet();
    void Init();
    int Find(int x);// 查找x的代表元素（根），查找的同时进行路径压缩
    void Union(int x,int y);
    int GetAnswer();// 合并x和y
};

DisJoinSet::DisJoinSet(int n)
{
    this->n = n ;
    tree = new node [n+1];
    Init();
    ans = n;
}

void DisJoinSet ::Init()
{
    for(int i = 1 ;i <=n; i++)//顶点编号 0～n-1  或  1 ～ n都 行 
    {
        tree[i].parent = i;//双亲初始化指向自已
        tree[i].rank = 0;//秩初始化为0
        tree[i].date = i;//编号
        tree[i].relation = 0;    //i自己是一类，它的父节点此时就是它自己，属于同一类
    }
}

DisJoinSet::~DisJoinSet()
{
    delete []tree;
}

int DisJoinSet::Find(int x)//查找x的代表元素（根），查找的同时进行路径压缩
{
    int temp = tree[x].parent;// 将x父节点的下标存入temp
    if( x != tree[x].parent)//若双亲不是自已
    {
        tree[x].parent = Find(tree[x].parent);//递归在双亲中找x
        return tree[x].parent;// 返回根节点下标
    }
    return x;//若双亲是自已,返回x
}

void DisJoinSet::Union(int x,int y)
{
    int rootx = Find(x);// 找到下标为x的元素的根节点下标rootx
    int rooty = Find(y);// 找到下标为y的元素的根节点下标rooty
    if(rootx == rooty)// 已合并，直接返回
    {
        return ;
    }
    else
    {
        if(tree[rootx].rank > tree[rooty].rank)//rooty结点的秩（深度）小于rootx结点的秩
        {
            tree[rooty].parent = rootx;//将rooty连到rootx结点上,rootx作为rooty的孩子结点
            ans--;//合并两个人，总人数-1
            tree[rootx].total +=tree[rooty].total;
        }
        else
        {
            tree[rootx].parent = rooty;
            ans--;//合并两个人，总人数-1
            tree[rooty].total += tree[rootx].total;
            if(tree[rootx].rank == tree[rooty].rank)
            {
                tree[rooty].rank++;
            }
        }
    }
}

int main()
{
    int n;
    int m;
    int t = 1;
    while(cin>>n>>m && n !=0 &&m != 0)
    {
        DisJoinSet dis(n);
        int a,b;
        for(int i = 1 ;i <= m; i++)
        {
            cin>>a>>b;
            dis.Union(a,b);
        }
        cout<<"Case "<<t++<<": "<<dis.ans<<endl;
    }
    return 0;
}