LeetCode 839. 相似字符串组（并查集）

示例 1：
输入：strs = ["tars","rats","arts","star"]
输出：2

示例 2：
输入：strs = ["omv","ovm"]
输出：1
 
提示：
1 <= strs.length <= 100
1 <= strs[i].length <= 1000
sum(strs[i].length) <= 2 * 10^4
strs[i] 只包含小写字母。
strs 中的所有单词都具有相同的长度，且是彼此的字母异位词。
 

备注：
字母异位词（anagram），一种把某个字符串的字母的位置（顺序）
加以改换所形成的新词。

class Solution {
    vector<int> f;
    int uni;
    int find(int a)
    {
        if(a == f[a]) return a;
        return f[a] = find(f[a]);
    }
    void merge(int a, int b)
    {
        int fa = find(a), fb = find(b);
        if(fa != fb)
        {
            f[fa] = fb;
            uni--;
        }
    }
public:
    int numSimilarGroups(vector<string>& strs) {
        int n = strs.size();
        uni = strs.size();
        f.resize(n);
        for(int i = 0; i < n; ++i)
            f[i] = i;
        for(int i = 0; i < n; ++i)
        {
            for(int j = i+1; j < n; ++j)
            {
                if(find(i) != find(j) && similar(strs[i], strs[j]))
                {
                    merge(i, j);
                }
            }
        }
        return uni;
    }
    bool similar(string& a, string& b)
    {
        int diff = 0;
        for(int i = 0; i < a.size(); ++i)
        {
            if(a[i] != b[i])
                diff++;
            if(diff > 2)
                return false;
        }
        return true;
    }
};