计算旋转字符串之间的相似度-Python教程-PHP中文网

计算旋转字符串之间的相似度

碧海醫心

发布： 2025-08-03 19:02:11

原创

861人浏览过

计算旋转字符串之间的相似度

本文介绍了一种高效的方法来判断字符串列表中哪些字符串是彼此旋转后的变体。通过将一个字符串复制两遍，然后检查另一个字符串是否是它的子串，可以有效地识别旋转后的相似字符串。此外，还提供了优化策略，例如按长度和字符计数预先分组字符串，以提高处理大量数据时的效率。

旋转字符串相似度判断的核心技巧

判断两个字符串 s1 和 s2 是否是彼此旋转的结果，关键在于检查 s1 是否是 2*s2 的子串。这个技巧的核心在于，如果 s1 是 s2 旋转得到的，那么 s1 一定包含在 s2 拼接自身的结果中。

Python 实现示例

以下代码展示了如何使用这个技巧来查找字符串列表中所有旋转相似的字符串：

data = ['1010', '122a916b2a916b7667110161', '20000000', '2020', '2a916b',
        '2a916b7667110161122a916b', '6b7667110161122a916b2a91', 
        '916b7667110161122a916b2a', 'ffff']

d = {}
for s in data:
    for k in d:
        if len(s) == len(k) and s in 2*k:
            d[k].append(s)
            break
    else:
        d[s] = [s]

result = [g for g in d.values() if len(g) > 1]
print(result)
# [['122a916b2a916b7667110161', '2a916b7667110161122a916b', '6b7667110161122a916b2a91', '916b7667110161122a916b2a']]

登录后复制

代码解释：

美间AI

美间AI：让设计更简单

查看详情

初始化： 创建一个空字典 d 用于存储分组结果。
外层循环： 遍历字符串列表 data 中的每个字符串 s。
内层循环： 遍历字典 d 中的每个键 k（代表一个已知的字符串组）。
条件判断：
- len(s) == len(k)：首先检查 s 和 k 的长度是否相等。如果长度不同，则它们不可能互为旋转字符串，直接跳过。
- s in 2*k：如果长度相等，则检查 s 是否是 2*k 的子串。如果是，则说明 s 是 k 的一个旋转变体。
分组： 如果 s 是 k 的旋转变体，则将 s 添加到 d[k] 对应的列表中，并跳出内层循环。
创建新组： 如果内层循环没有找到匹配的组，则说明 s 是一个新的字符串组的代表，创建一个新的键值对 d[s] = [s]。
提取结果： 最后，提取字典 d 中所有长度大于 1 的列表，这些列表包含了所有旋转相似的字符串组。

性能优化

当处理大量数据时，直接进行字符串匹配可能会比较耗时。可以通过以下方法进行优化：

按长度分组： 预先将字符串按照长度进行分组，只有长度相同的字符串才需要进行旋转检查。
按字符计数分组： 在长度相同的基础上，进一步按字符计数进行分组。只有字符计数相同的字符串才有可能互为旋转字符串。可以使用 collections.Counter 来实现字符计数。

示例代码（包含长度和字符计数分组）：

from collections import Counter

data = ['1010', '122a916b2a916b7667110161', '20000000', '2020', '2a916b',
        '2a916b7667110161122a916b', '6b7667110161122a916b2a91', 
        '916b7667110161122a916b2a', 'ffff']

# 按长度和字符计数分组
groups = {}
for s in data:
    key = (len(s), tuple(sorted(Counter(s).items())))  # 使用tuple避免Counter不可哈希的问题
    if key not in groups:
        groups[key] = []
    groups[key].append(s)

result = []
for group in groups.values():
    if len(group) > 1:
        d = {}
        for s in group:
            for k in d:
                if s in 2*k:
                    d[k].append(s)
                    break
            else:
                d[s] = [s]
        result.extend([g for g in d.values() if len(g) > 1])

print(result)
# [['122a916b2a916b7667110161', '2a916b7667110161122a916b', '6b7667110161122a916b2a91', '916b7667110161122a916b2a']]

登录后复制