hihocoder1066 并查集

并查集可实现集合的快速合并与查找操作。路径压缩后的并查集可将每次合并或者查找的操作复杂度降低到O(1).

我的代码：

#include <iostream>
#include <string>
#include <map>

using namespace std;

#define MAXN 100005

map<string, int> mp;

struct unionFindSet
{
    int st[MAXN];
    void init()
    {
        for(int i=0; i<MAXN; ++i)   st[i] = i;
    }
    int findSet(int x)
    {
        if(x==st[x]) return x;
        return st[x] = findSet(st[x]); //路径压缩
    }
    void unionSet(int x, int y)
    {
        int sx = findSet(x), sy = findSet(y);
        st[sx] = sy;
    }
}ufs;

int main()
{
    int cnt, n;
    bool op;
    string name1, name2;
    while(cin>>n)
    {
        mp.clear();
        cnt = 0;
        ufs.init();
        while(n--)
        {
            cin>>op>>name1>>name2;
            if(mp.find(name1)==mp.end())    mp[name1] = cnt++;
            if(mp.find(name2)==mp.end())    mp[name2] = cnt++;
            if(op)
            {
                int s1 = ufs.findSet(mp[name1]), s2 = ufs.findSet(mp[name2]);
                cout<<(s1==s2?"yes":"no")<<endl;
            }
            else
            {
                ufs.unionSet(mp[name1], mp[name2]);
            }
        }
    }
    return 0;
}

题目链接：http://hihocoder.com/problemset/problem/1066