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

2022/02/26 leetcode 共 902 字,约 3 分钟

title

problem link

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

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

本题将会给你一个字符串 s ,我们需要返回 countUniqueChars(t) 的总和,其中 t 是 s 的子字符串。注意,某些子字符串可能是重复的,但你统计时也必须算上这些重复的子字符串(也就是说,你必须统计 s 的所有子字符串中的唯一字符)。

由于答案可能非常大,请将结果 mod 10 ^ 9 + 7 后再返回。

示例 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

提示:

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

solution

统计每个字符在每个子串中唯一出现时的次数,其实就是就是维护last index, 然后左边数量乘以右边数量就是当前字符c在每个子串中出现次数。

对每个字符两边推展,计算每个字母两边可以最长的独特次数

mod那个实际并没有用到,可以加。

class Solution:
    def uniqueLetterString(self, s: str) -> int:
        n = len(s)
        left = {}
        right = [{} for _ in range(n + 1)]
        for i in range(n - 1, -1, -1):
            right[i] = right[i + 1].copy()
            c = s[i]
            right[i][c] = i
        ans = 0
        for i, c in enumerate(s):
            a = i - left.get(c, -1)
            b = right[i + 1].get(c, n)-i
            # print(i, a, b)
            ans += a * b
            left[c] = i
        return ans

Search

    Table of Contents