Leetcode每日一题：839. 相似字符串组

最新推荐文章于 2022-03-10 20:00:26 发布

不会写代码的嘤嘤

最新推荐文章于 2022-03-10 20:00:26 发布

阅读量114

点赞数 1

分类专栏： leetcode练习文章标签： leetcode python

本文链接：https://blog.csdn.net/qaqlalala/article/details/113475269

版权

leetcode练习专栏收录该内容

47 篇文章 0 订阅

订阅专栏

本文探讨如何利用并查集数据结构解决字符串列表中相似字符串组的问题。通过实例分析，展示了如何通过比较字符异位和连接相似组来计数有效组数。代码实现包括DSU类的构造和Solution类的numSimilarGroups方法。

摘要由CSDN通过智能技术生成

这周做了好几道并查集的题，对于并查集也有了一定的掌握，今天在做每日一题的时候也能大体有一个解题的思路

问题描述

如果交换字符串 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），一种把某个字符串的字母的位置（顺序）加以改换所形成的新词。

思路分析及代码实现

1.套用并查集模板
2.两个for循环遍历所有点
3.定义一个count，遍历两点的对应位置是否相同，如果不相同那么count+1
4.判断count是否等于0或者2，如果等于则将两点连接，同时不连通区域数量-1

class DSU:
    def __init__(self, n):
        self.father = [i for i in range(n)]
        self.num_set = n
    
    def find(self, x):
        if x != self.father[x]:
            self.father[x] = self.find(self.father[x])
        return self.father[x]
    
    def is_connect(self, x, y):
        return self.find(x) == self.find(y)
    
    def union(self, x, y):
        root_x, root_y = self.find(x), self.find(y)
        if root_x != root_y:
            self.father[root_y] = root_x
            self.num_set -= 1

class Solution:
    def numSimilarGroups(self, strs: List[str]) -> int:
        n = len(strs)
        m = len(strs[0])
        dsu = DSU(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]:
                    dsu.union(i, j)
        return dsu.num_set