Implementation：UnionFindSet 并查集

class UnionFindSet {
    private:
        int *pref;
        int *rank;
        int capacity;
    public:
        UnionFindSet(int n) {
            if (n <= ) {
                capacity = ;
                return;
            }
            capacity = n;
            pref = new int[n];
            rank = new int[n];
            for (int i=; i<n; i++) {
                rank[i] = ;
                pref[i] = i;
            }
        }

        int find(int s) {
            if (s >= capacity || s < ) return -;
            if (s == pref[s]) return s;
            return pref[s] = find(pref[s]);
        }

        void unite(int a, int b) {
            if (a >= capacity || b >= capacity || a <  || b < ) return;
            int pa = find(a), pb = find(b);
            if (pa == pb) return; // already united
            if (rank[pa] < rank[pb]) {
                pref[pa] = pb;
            } else {
                pref[pb] = pa;
            }
            if (rank[pa] == rank[pb]) rank[pa]++;
        }

        bool same(int a, int b) {
            return find(a) == find(b);
        }
};

并查集，练习一下