每日一练 2024.9.29（2）-CSDN博客

class Solution {
    // 并查集数组
    int[] parent;

    // 查找操作
    public int find(int x) {
        if (parent[x] != x) {
            parent[x] = find(parent[x]); // 路径压缩
        }
        return parent[x];
    }

    // 合并操作
    public void union(int x, int y) {
        int rootX = find(x);
        int rootY = find(y);
        if (rootX != rootY) {
            parent[rootX] = rootY; // 合并两个集合
        }
    }

    // 判断两个字符串是否相似
    public boolean isSimilar(String a, String b) {
        int diff = 0;
        for (int i = 0; i < a.length(); i++) {
            if (a.charAt(i) != b.charAt(i)) {
                diff++;
                if (diff > 2) {  // 如果超过2个位置不同，则不相似
                    return false;
                }
            }
        }
        return true;
    }

    public int numSimilarGroups(String[] strs) {
        int n = strs.length;
        parent = new int[n];

        // 初始化并查集，每个节点是自己的父节点
        for (int i = 0; i < n; i++) {
            parent[i] = i;
        }

        // 进行合并操作
        for (int i = 0; i < n; i++) {
            for (int j = i + 1; j < n; j++) {
                if (isSimilar(strs[i], strs[j])) {
                    union(i, j);
                }
            }
        }

        // 统计连通分量的数量
        int count = 0;
        for (int i = 0; i < n; i++) {
            if (find(i) == i) {  // 如果是根节点，则表示是一个新的字符组
                count++;
            }
        }

        return count;
    }
}

三、代码解析

并查集初始化：parent[i] = i，表示每个节点一开始都是独立的组，自己是自己的父节点。
相似性判断：通过遍历字符串的每个字符，统计不同字符的个数，如果超过2个字符不相同，则它们不相似。
合并集合：通过 union(i, j) 来合并两个相似的字符串。
查找连通分量：最终通过查找根节点，统计有多少个独立的组（即多少个相似字符组）。

四、复杂度分析

时间复杂度：
- 并查集的初始化为 O(n)，其中 n 是字符串的数量。
- 双重循环判断字符串的相似性，复杂度为 O(n^2 * m)，其中 m 是字符串的长度，因为每次比较字符串相似性需要 O(m) 的时间。
- 合并和查找的操作接近常数时间，路径压缩可以使得查找的时间复杂度接近于 O(1)。
- 总的时间复杂度为 O(n^2 * m)。
空间复杂度：
- 主要空间开销是并查集的数组 parent，占用 O(n) 的空间。
- 额外的空间开销较少，因此总的空间复杂度为 O(n)。

五、运行示例

示例1：

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

解释：

"tars" 和 "rats" 相似（交换1次）。
"rats" 和 "arts" 相似（交换1次）。
"arts" 和 "star" 相似（交换1次）。所以这四个字符串形成一个相似字符组，答案是 1。

示例2：

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

解释：

"omv" 和 "ovm" 相似（交换1次）。所以它们属于同一组，答案是 1。

六、总结

在这道题中，我们通过并查集结构实现了对相似字符组的判断和合并操作。通过设计相似性判断函数，我们能够准确地将相似的字符串分为一组。最终，通过并查集的查找操作，统计出有多少个相似字符组。通过这种方法，我们解决了字符串相似性分组的问题，并且该算法具有较好的时间和空间效率。