UnionFindのテンプレコード(C++)の実装

namespace template_algorithm {
    class union_find {
        public:
        /// @brief コンストラクタ
        union_find(int n) {
            roots.assign(n, Pi());
            for (int i = 0; i < (int)roots.size(); i++) {
                roots[i] = {i, 1};
            }
        }

        /// @brief aとbがそれぞれ含まれる、部分木同士を結合する
        void unite(int a, int b) {
            a = get_root(a);
            b = get_root(b);
            //すでに同じ木に含まれる場合
            if (a == b) return;
            //aの部分木のほうが大きかった場合
            if (roots[a].second > roots[b].second) {
                roots[b].first = a;
                roots[a].second += roots[b].second;
            }
            //bの部分木のほうが大きかった場合
            else {
                roots[a].first = b;
                roots[b].second += roots[a].second;
            }
        }

        /// @brief ノードa, bが同じ部分木に含まれるかの判定
        bool is_same_root(int a, int b) {
            a = get_root(a);
            b = get_root(b);
            return a == b;
        }

        /// @brief xの根ノードを返す
        int get_root(int x) {
            if (roots[x].first == x) return x;
            roots[x].first = get_root(roots[x].first);
            roots[x].second = roots[roots[x].first].second;
            return roots[x].first;
        }

        /// @brief xが含まれる部分木のサイズを返す
        int get_tree_size(int x) {
            x = get_root(x);
            return roots[x].second;
        }

        private:
        using Pi = pair<int, int>;
        //first:根ノード second:部分木のサイズ
        vector<Pi> roots;
    };
}