力扣题解-839. 相似字符串组（并查集）

题目：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

提示：

• 1 <= strs.length <= 100
• 1 <= strs[i].length <= 1000
• sum(strs[i].length) <= 2 * 104
• strs[i] 只包含小写字母。
• strs 中的所有单词都具有相同的长度，且是彼此的字母异位词。

备注：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/similar-string-groups
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题解

题目的意思是，只要字符串a与一个字符串组的任意一个字符串是相似的，就可以将这个字符串合并到该字符串组中。

换句话说，一个字符串组中的每一个字符串只要与其中一个其他的字符串是相似的。

因此，可以利用并查集来解决字符串的归类问题。

我们遍历全部字符串，并两两进行相似性比较，成功则合并，最后得到的并查集的集合个数即为相似字符串组的个数。

代码

class UnionFind {
private:
    int n;
    vector<int> parent;
public:
    UnionFind(int n_): n(n_) {
        parent.resize(n);
        for(int i = 0; i < n; i++) {
            parent[i] = i;
        }
    }

    int find(int x) {
        int root = x;
        while(root != parent[root]) {
            root = parent[root];
        }

        while(x != parent[x]) {
            int tmp = parent[x];
            parent[x] = root;
            x = tmp;
        }
        return root;
    }

    bool merge(int x, int y) {
        int px = find(x);
        int py = find(y);

        if(px == py) {
            return false;
        }
        parent[py] = px;
        n--;
        return true;
    }
    int getSetNum(void) {
        return n;
    }
};

class Solution {
public:
    int numSimilarGroups(vector<string>& strs) {
        int n = strs.size();
        UnionFind unionFind(n);
        int length = strs[0].length();
        for(int i = 0; i < n; i++) {
            for(int j = i + 1; j < n; j++) {
                if(isSimilar(strs[i], strs[j], length) == true) {
                    unionFind.merge(i, j);
                }
            }
        }
        return unionFind.getSetNum();
    }
    bool isSimilar(string& str1, string& str2, int length) {
        int count = 0;
        for(int i = 0; i < length; i++) {
            if(str1[i] != str2[i]) {
                count++;
            }
        }
        return count > 2 ? false: true;
    }
};