LeetCode每日一题：groupby()函数

1759. 统计同构子字符串的数目

给你一个字符串 s ，返回 s 中 同构子字符串 的数目。由于答案可能很大，只需返回对 109 + 7 取余 后的结果。
同构字符串 的定义为：如果一个字符串中的所有字符都相同，那么该字符串就是同构字符串。
子字符串 是字符串中的一个连续字符序列。

输入：s = "abbcccaa"
输出：13
解释：同构子字符串如下所列：
"a" 出现 3 次。
"aa" 出现 1 次。
"b" 出现 2 次。
"bb" 出现 1 次。
"c" 出现 3 次。
"cc" 出现 2 次。
"ccc" 出现 1 次。
3 + 1 + 2 + 1 + 3 + 2 + 1 = 13

class Solution:
    def countHomogenous(self, s: str) -> int:
        res = 0
        for k, g in groupby(s):
            # print(k,list(g))
            n = len(list(g))
            res += (n + 1)* n // 2
        return res % (10 ** 9 + 7)

• 关于groupby()函数

s = "abbcccaa"
for k, g in groupby(s):
    print(k,list(g))

a ['a']
b ['b', 'b']
c ['c', 'c', 'c']
a ['a', 'a']

函数将字符串中相邻的字符串进行分组