【LeetCode】828. 统计子串中的唯一字符

题目

828. 统计子串中的唯一字符

我们定义了一个函数 countUniqueChars(s) 来统计字符串 s 中的唯一字符，并返回唯一字符的个数。

例如：s = "LEETCODE" ，则其中 "L", "T","C","O","D" 都是唯一字符，因为它们只出现一次，所以 countUniqueChars(s) = 5 。

本题将会给你一个字符串 s ，我们需要返回 countUniqueChars(t) 的总和，其中 t 是 s 的子字符串。输入用例保证返回值为 32 位整数。

注意，某些子字符串可能是重复的，但你统计时也必须算上这些重复的子字符串（也就是说，你必须统计 s 的所有子字符串中的唯一字符）。

示例 1：

输入: s = "ABC"
输出: 10
解释: 所有可能的子串为："A","B","C","AB","BC" 和 "ABC"。
     其中，每一个子串都由独特字符构成。
     所以其长度总和为：1 + 1 + 1 + 2 + 2 + 3 = 10

示例 2：

输入: s = "ABA"
输出: 8
解释: 除了 countUniqueChars("ABA") = 1 之外，其余与示例 1 相同。

示例 3：

输入：s = "LEETCODE"
输出：92

提示：

• 1 <= s.length <= 10^5
• s 只包含大写英文字符

思路

• 保存每个字符出现的下标，记为index则，在index[i-1]道index[i+1]的区间当中，这个字母只出现过一次
• 当字母最后一次出现，或第一次出现，无匹配，所以需要再index前后补充-1，n

代码

from collections import defaultdict
class Solution:
    def uniqueLetterString(self, s: str) -> int:
        indices = defaultdict(list)
        ret = 0
        for i,c in enumerate(s):
            indices[c].append(i)
        for index in indices.values():
            index = [-1] + index + [len(s)]
            for i in range(1, len(index) - 1):
                ret += (index[i+1] - index[i]) * (index[i] - index[i-1])
        return ret

复杂度

• 时间复杂度：$O(n)$
• 空间复杂度：$O(n)$