相似字符串组

/*
 * 思路：
 * 1. 使用流 (Stream) 的方式来简化代码，尤其是在判断相似性和组的合并过程中。
 * 2. 使用并查集(Union-Find)来管理相似组。
 * 3. 最后统计不同的组数。
 */

import java.util.stream.IntStream;

class Solution {
    private int find(int[] parent, int i) {
        if (parent[i] != i) {
            parent[i] = find(parent, parent[i]);
        }
        return parent[i];
    }

    private void union(int[] parent, int[] rank, int i, int j) {
        int rootI = find(parent, i);
        int rootJ = find(parent, j);
        if (rootI != rootJ) {
            if (rank[rootI] > rank[rootJ]) {
                parent[rootJ] = rootI;
            } else if (rank[rootI] < rank[rootJ]) {
                parent[rootI] = rootJ;
            } else {
                parent[rootJ] = rootI;
                rank[rootI]++;
            }
        }
    }

    private boolean areSimilar(String a, String b) {
        return IntStream.range(0, a.length())
            .filter(i -> a.charAt(i) != b.charAt(i))
            .count() <= 2;
    }

    public int numSimilarGroups(String[] strs) {
        int n = strs.length;
        int[] parent = IntStream.range(0, n).toArray();
        int[] rank = new int[n];

        IntStream.range(0, n).forEach(i -> 
            IntStream.range(i + 1, n).filter(j -> areSimilar(strs[i], strs[j]))
            .forEach(j -> union(parent, rank, i, j))
        );

        return (int) IntStream.range(0, n).filter(i -> parent[i] == i).count();
    }
}