题目
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),一种把某个字符串的字母的位置(顺序)加以改换所形成的新词。
解题思路
- 这道题感觉说的有点懵,但是读了很久之后发现了这道题的实质。
- 我们通过题意可以理解到相似字符串组的含义,就是把每个单词当作节点,如果两个节点是相似的,那么他们就连通在一起,否则就不连通。
- 很显然,这道题又双叒叕(yòu,shuāng,ruò,zhuó”)用到了并查集的方法。
- 并查集——就是把每个单词作为节点,遍历一遍,看有多少个节点是不连通的,不连通的数目就是答案。
Code
#并查集模板,计算连通分量
class UnionFind:
def __init__(self, x):
self.father = [i for i in range(x)]
self.count = x
def find(self, x):
if self.father[x] == x:
return x
self.father[x] = self.find(self.father[x])
return self.father[x]
def union(self, x, y):
x, y = self.find(x), self.find(y)
if x != y:
self.father[x] = y
self.count -= 1
class Solution:
def numSimilarGroups(self, strs: List[str]) -> int:
n, m = len(strs), len(strs[0])
uf = UnionFind(n)
for i in range(n):
for j in range(i + 1, n):
count = 0
for k in range(m):
if strs[i][k] != strs[j][k]:
count += 1
if count > 2:
break
if count in {0, 2}:
uf.union(i, j)
return uf.count
运行结果