839. 相似字符串组

题目描述：如果交换字符串 X 中的两个不同位置的字母，使得它和字符串 Y 相等，那么称 X 和 Y 两个字符串相似。如果这两个字符串本身是相等的，那它们也是相似的。
例如，“tars” 和 “rats” 是相似的 (交换 0 与 2 的位置)； “rats” 和 “arts” 也是相似的，但是 “star” 不与 “tars”，“rats”，或 “arts” 相似。
总之，它们通过相似性形成了两个关联组：{“tars”, “rats”, “arts”} 和 {“star”}。注意，“tars” 和 “arts” 是在同一组中，即使它们并不相似。形式上，对每个组而言，要确定一个单词在组中，只需要这个词和该组中至少一个单词相似。
给你一个字符串列表 strs。列表中的每个字符串都是 strs 中其它所有字符串的一个字母异位词。请问 strs 中有多少个相似字符串组？
解题思路：将每个字符作为一个节点，如果两个字符相似，那么对应两个节点之间连有一条无向边，按照这个规则构建一个图，然后利用并查集进行节点的合并，最后统计集合的个数，代码如下：

class Solution:
    def numSimilarGroups(self, strs: List[str]) -> int:
        edges = []
        n = len(strs)
        def issim(s1, s2):
            count = 0
            for a, b in zip(s1, s2):
                if a != b:
                    count += 1
                    if count > 2:
                        return False
            return True
        for i in range(n):
            for j in range(i+1, n):
                if issim(strs[i], strs[j]):
                    edges.append([i, j])
        father = list(range(n))
        def find(x):
            root = x
            while(root != father[root]):
                root = father[root]
            while(x != root):
                origin_r = father[x]
                father[x] = root
                x = origin_r
            return root
        def union(x, y):
            rx, ry = find(x), find(y)
            if rx != ry:
                father[ry] = rx
            
        for u, v in edges:
            union(u, v)
        return len(set([find(f) for f in father]))