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']
函數將字符串中相鄰的字符串進行分組