leetcode_1180 统计只含单一字母的子串

337 阅读1分钟

要求

给你一个字符串 S,返回只含 单一字母 的子串个数。

示例 1:

输入: "aaaba"
输出: 8
解释: 
只含单一字母的子串分别是 "aaa""aa""a""b""aaa" 出现 1 次。
"aa" 出现 2 次。
"a" 出现 4 次。
"b" 出现 1 次。
所以答案是 1 + 2 + 4 + 1 = 8

示例 2:

输入: "aaaaaaaaaa"
输出: 55

提示:

  • 1 <= S.length <= 1000
  • S[i] 仅由小写英文字母组成。

核心代码

class Solution:
    def countLetters(self, s: str) -> int:
        res = 0
        for _,j in itertools.groupby(s):
            t = len(list(j))
            res += (1 + t) * t // 2
        return res

image.png

解题思路:我们使用itertools.groupby这个函数,我们可以发现"AAABBBCCAAA",最终得到的是"AAA","BBB","CC","AAA",我们,这是一个找规律的问题,规律很简单,就是统计字符串中连续相同字符的个数,如"aaaba"的输出是311。接着对每个数采用等差数列求和即可,(1+3)*3//2 + (1+1)*1//2+(1+1)*1//2=8