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