非交差セットクラス

4170 ワード

等価関係とうかかんけい:自己反転,対称,伝達
class DisjSets// 
{
public:
    explicit DisjSets(int numElements);
    int find(int x) const;
    int find(int x);
    void unionSets(int root1,int root2);
    void unionSets2(int root1,int root2);
private:
    vector<int> s;
};
DisjSets::DisjSets(int numElements) : s(numElements)// 
{
    for(int i=0 ; i < s.size() ; i++)
        s[i] = -1;
}
void DisjSets::unionSets(int root1,int root2)
{
    s[root2] = root1;
}
void DisjSets::unionSets2(int root1,int root2)
{
    if( s[root2] < s[root1])
        s[root1] = root2;
    else
    {
        if(s[root1] == s[root2])
            s[root1]--;
        s[root2] = root1;
    }
}
int DisjSets::find(int x) const
{
    if(s[x] < 0)
        return x;
    else
        return find(s[x]);
}

コンパクト集計アルゴリズム:サイズ別集計、または高さ別集計
パス圧縮:唯一の変化はfindが返す値(サイズで求めて完全に互換性がある)を返すことです.
int DisjSets::find(int x) 
{
    if(s[x] < 0)
        return x;
    else
        return s[x] = find(s[x]);
}

応用:迷宮の問題