LeetCode 839 相似字符串组

题目链接

如果交换字符串 X 中的两个不同位置的字母，使得它和字符串 Y 相等，那么称 X 和 Y 两个字符串相似。如果这两个字符串本身是相等的，那它们也是相似的。

例如，”tars” 和 “rats” 是相似的 (交换 0 与 2 的位置)； “rats” 和 “arts” 也是相似的，但是 “star” 不与 “tars”，”rats”，或 “arts” 相似。

总之，它们通过相似性形成了两个关联组：{“tars”, “rats”, “arts”} 和 {“star”}。注意，”tars” 和 “arts” 是在同一组中，即使它们并不相似。形式上，对每个组而言，要确定一个单词在组中，只需要这个词和该组中至少一个单词相似。

给你一个字符串列表 strs。列表中的每个字符串都是 strs 中其它所有字符串的一个字母异位词。请问 strs 中有多少个相似字符串组？

示例 1：

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

示例 2：

输入：strs = ["omv","ovm"]
输出：1

并查集~
双重循环遍历字符串组，把相似的字符串合并到同一个并查集里，最后找一下并查集里连通块的个数即可，AC代码如下：

class Solution {
public:
    vector<int> f;

    int findFather(int x) {
        return x == f[x] ? x : f[x] = findFather(f[x]);
    }

    bool check(string &a, string &b, int len) {
        int cnt = 0;
        for (int i = 0; i < len; i++) {
            if (a[i] != b[i]) {
                cnt++;
                if (cnt > 2) return 0;
            }
        }
        return 1;
    }

    int numSimilarGroups(vector<string> &strs) {
        int n = strs.size();
        if (n == 0) return 0;
        f.resize(n);
        for (int i = 0; i < n; i++) f[i] = i;
        int len = strs[0].size(), ans = 0;
        for (int i = 0; i < n; i++) {
            for (int j = i + 1; j < n; j++) {
                int fi = findFather(i), fj = findFather(j);
                if (fi == fj) continue;
                if (check(strs[i], strs[j], len)) {
                    f[fi] = fj;
                }
            }
        }
        for (int i = 0; i < n; i++) if (f[i] == i) ans++;
        return ans;
    }
};