终于最后一道题了,由于这个题真的试了好多次都不能过测试,so 我就先去写后面的了,回来把这题给再补上,ok,下面就是本次初赛的唯二困难题,也是最后一题,相似字符串组
T18 如果交换字符串 X 中的两个不同位置的字母,使得它和字符串 Y 相等,那么称 X 和 Y 两个字符串相似。如果这两个字符串本身是相等的,那它们也是相似的。
例如,“tars” 和 “rats” 是相似的 (交换 0 与 2 的位置); “rats” 和 “arts” 也是相似的,但是 “star” 不与 “tars”,“rats”,或 “arts” 相似。
总之,它们通过相似性形成了两个关联组:{“tars”, “rats”, “arts”} 和 {“star”}。注意,“tars” 和 “arts” 是在同一组中,即使它们并不相似。形式上,对每个组而言,要确定一个单词在组中,只需要这个词和该组中至少一个单词相似。
给你一个字符串列表 strs。列表中的每个字符串都是 strs 中其它所有字符串的一个字母异位词。请问 strs 中有多少个相似字符串组?
解题思路:
刚开始其实我一直通过逻辑关系,逐渐一步步递推,到后来发现就是拆了东墙补西墙,理的脑子疼也没修好bug,看了下题解才知道有并查集这种东西,接下来我讲下自己的理解。
相似字符串组其实和之前做的最大省个数类似,将每个数看作是点的话,两点相似就代表这两个点相连通,共为一群,我们需要求出最大的群的个数。
并查集实质上就是建立一个包含所有点的列表,列表的index代表该点,数值则代表该点的上一连接。为了方便理解,我们这里举个例子,结合代码应该更容易看懂
假设有0,1,2三点,0分别与1、2相连,初始表d为[ 0, 1, 2 ],代表0连0,1连1,2连2。遍历,首先发现0与1相连,则初始表d变为[1 ,1 , 2],表示0连1,1连1,2连2; 之后又找到0与2相连,d更新为[1, 2, 2],表示0连1,1连2,2连2;在这里,2就代表最上级,其下一级是1,再下一级是0,全部遍历完毕后,仍保留为原本数值的为所有的最上级,其个数就是所有的群数,代码如下:
class Solution:
def numSimilarGroups(self, strs: List[str]) -> int:
def is_similar(a,b):
test = 0
for i in range(len(a)):
if a[i] != b[i]:
test += 1
if test > 2:
return False
return True
def find(d, x):
if d[x] == x:
return x
else:
return find(d, d[x])
d = [i for i in range(len(strs))]
for i in range(len(strs)):
for j in range(i+1, len(strs)):
di = find(d, i)
dj = find(d, j)
if is_similar(strs[i],strs[j]):
d[di] = dj
res = 0
for i in range(len(strs)):
if d[i] == i:
res += 1
return res
运行结果:
再把自己想的逻辑一点点推的方法附上,yysy有bug,所以这里就不再多讲了,把它写上就权当代表我徒劳的一上午吧,大家用上面那一个即可
class Solution:
def numSimilarGroups(self, strs: List[str]) -> int:
def is_similar(a,b):
test = 0
for i in range(len(a)):
if a[i] != b[i]:
test += 1
if test == 2 or test == 0:
return True
else:
return False
used = [False]*len(strs)
test = [[] for _ in range(len(strs))]
res = 0
f = 0
for i in range(len(strs)):
for j in range(i+1,len(strs)):
for t in test:
if i in t and j in t:
f = 1
break
if f == 1:
f = 0
continue
if is_similar(strs[i], strs[j]):
if used[i] and used[j]:
res -= 1
if not used[i] and not used[j]:
res += 1
used[i] = True
used[j] = True
test[i].append(j)
for i in used:
if not i:
res += 1
if res < 0 :
return 1
return res