[Swift]DJSet

★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★
➤微信公众号：山青咏芝（shanqingyongzhi）
➤博客园地址：山青咏芝（https://www.cnblogs.com/strengthen/ ）
➤GitHub地址：https://github.com/strengthen/LeetCode
➤原文地址：https://www.cnblogs.com/strengthen/p/10053038.html 
➤如果链接不是山青咏芝的博客园地址，则可能是爬取作者的文章。
➤原文已修改更新！强烈建议点击原文地址阅读！支持作者！支持原创！
★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★

public class DJSet
{
    var upper:[Int]
    var w:[Int]
    
    init(_ n:Int)
    {
        upper = [Int](repeating:-1,count:n)
        w = [Int](repeating:0,count:n)
    }
    
    func root(_ x:Int) -> Int
    {
        if(upper[x] < 0)
        {
            return x
        }
        else
        {
            upper[x] = root(upper[x])
            return upper[x]
        }
    }
    
    func equiv(_ x:Int,_ y:Int) -> Bool
    {
        return root(x) == root(y)
    }
    
    func union(_ x:Int,_ y:Int) -> Bool
    {
        var x:Int = root(x)
        var y:Int = root(y)
        if x != y
        {
            if upper[y] < upper[x]
            {
                var d:Int = x
                x = y
                y = d
            }
            upper[x] += upper[y]
            upper[y] = x
            w[x] += w[y]
        }
        return x == y
    }
    
    func count() -> Int
    {
        var ct:Int = 0
        for u in upper
        {
            if u < 0
            {
                ct += 1
            }
        }
        return ct
    }
}