poj 2524 Ubiquitous Religions （STL与非STL的对比）

http://poj.org/problem?id=2524

　　这不是一道难题，只是一个普通的，相当基础的MFSet问题。

　　这题的题解到处都有，所以我就只是大概说说做法：将编号按照关系分类，最后统计一共有子树。

　　我提供了两种代码，主要是为了说明STL消耗较多的内存和时间。下面的是用了set的，明显内存消耗更大：

　　下面是我的代码，只要调整STL的值就可以改变统计方法：

#include "iostream"
#include "cstdio"
#include "cstdlib"
#include "cstring"
#include "cmath"
#include "cctype"
#include "vector"
#include "set"
#include "map"
#include "string"
#include "algorithm"
#include "stack"
#include "utility"
#include "queue"
using namespace std;

#define INF 0x7fffffff
#define reset(a) memset(a, 0, sizeof(a))
#define copy(a, b) memcpy(a, b, sizeof(b))
#define PI acos(-1)
#define FMAX (1E300)
#define MAX 1000000000
#define feq(a, b) (fabs((a)-(b))<1E-6)
#define flq(a, b) ((a)<(b)||feq(a, b))
#define MAXN 50005

#define STL 1

int f[MAXN], cnt[MAXN];

void init_set(int n){
    for (int i=0; i<=n; i++)
        f[i]=i, cnt[i] = 1;
}

int find_root(int p){
    if (f[p]!=p)
        f[p] = find_root(f[p]);
    return f[p];
}

void merge_set(int p, int q){
    int a = find_root(p), b = find_root(q);
    if (cnt[a] < cnt[b])
        f[a] = b, cnt[b] += cnt[a];
    else
        f[b] = a, cnt[a] += cnt[b];
}

void fix_set(int n){
    for (int i=0; i<=n; i++)
        f[i] = find_root(i);
}

int main(){
    int m, n, CASE=0;

    while (~scanf("%d%d", &n, &m) && (n || m)){
        init_set(n);
        while (m--){
            int a, b;

            scanf("%d%d", &a, &b);
            merge_set(a, b);
        }
#if STL
        set<int> s;
        s.clear();
        for (int i=1; i<=n; i++)
            s.insert(find_root(i));
        printf("Case %d: %d\n", ++CASE, s.size());
#else
        bool vis[MAXN];
        int ans=0;
        reset(vis);
        for (int i=1; i<=n; i++){
            int t = find_root(i);
            if (!vis[t])
                ans++;
            vis[t]=true;
        }
        printf("Case %d: %d\n", ++CASE, ans);
#endif
    }

    return 0;
}