给定字符串中相同字符出现之间所有距离的总和

在字符串处理中，有一个常见的需求是计算相同字符出现之间的距离总和。比如说，对于字符串 "abca"，相同字符 "a" 出现的位置分别为 0 和 3，它们之间的距离为 3，而相同字符 "b" 出现在位置 1，因此它与相同字符 "a" 之间的距离为 1。因此，相同字符出现之间的距离总和为 3 + 1 = 4。

本文将介绍一种基于哈希表的算法来计算相同字符出现之间所有距离的总和。

算法思路

1. 首先，创建一个哈希表来存储每个字符出现的位置。
2. 然后，遍历字符串中的每个字符，对于每个字符：
   • 如果该字符在哈希表中不存在，则将其加入哈希表中，并将其出现的位置记录下来。
   • 如果该字符在哈希表中已经存在，则计算其出现位置与上一次出现位置之间的距离，并将距离加入总和中，同时将该字符的出现位置更新为当前位置。
3. 最后，返回距离总和即可。

代码实现

def total_distance(s: str) -> int:
    pos = {}
    dist = 0
    for i, c in enumerate(s):
        if c in pos:
            dist += i - pos[c]
        pos[c] = i
    return dist

示例

>>> total_distance("abca")
4
>>> total_distance("abcba")
5
>>> total_distance("abccba")
6