【Leetcode】839. Similar String Groups

题目地址：

https://leetcode.com/problems/similar-string-groups/

给定一个长$n$的字符串数组$A$，题目保证每个单词长度相同，并且都是变位词（即字母构成完全一样）。如果两个单词相等，或者其中一个交换某两个字符之后就等于另一个了，则这两个单词属于同一组，并且“属于同一组”这个关系有传递性。问$A$被分为了多少组。

可以用并查集，将相等或者交换某两个字符之后就相等的字符串的下标union起来。“交换某两个字符之后就相等”可以这样判断，直接遍历两个字符串，统计对应位置有多少个字母不同，如果不同的数量大于$2$了则直接返回false，如果遍历完成之后都没有返回false，说明对应位置不同的只有少于等于$2$个，而$1$是不可能的，$0$代表相等，$2$代表交换某两个位置后相等，都可以返回true。代码如下：

public class Solution {
    
    class UnionFind {
        private int[] p;
        int group;
    
        public UnionFind(int size) {
            p = new int[size];
            for (int i = 0; i < size; i++) {
                p[i] = i;
            }
            group = size;
        }
        
        public int find(int x) {
            if (p[x] != x) {
                p[x] = find(p[x]);
            }
            
            return p[x];
        }
        
        public void union(int x, int y) {
            int px = find(x), py = find(y);
            if (px != py) {
                p[px] = py;
                group--;
            }
        }
    }
    
    public int numSimilarGroups(String[] strs) {
        UnionFind uf = new UnionFind(strs.length);
        for (int i = 0; i < strs.length; i++) {
            for (int j = i + 1; j < strs.length; j++) {
                if (areSimilar(strs[i], strs[j])) {
                    uf.union(i, j);
                }
            }
        }
        
        return uf.group;
    }
    
    private boolean areSimilar(String s, String t) {
        int cnt = 0;
        for (int i = 0; i < s.length(); i++) {
            if (s.charAt(i) != t.charAt(i)) {
                cnt++;
            }
            
            if (cnt > 2) {
                return false;
            }
        }
        
        return true;
    }
}

时间复杂度$O(n^{2}l{\log }^{\ast }n)$，$l$是单词长度，空间$O(n)$。