LeetCode-Python-1180. 统计只含单一字母的子串

暴躁老哥在线刷题

已于 2024-09-16 08:07:12 修改

阅读量979

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode 文章标签：数学字符串

于 2019-09-08 01:02:50 首次发布

本文链接：https://blog.csdn.net/qq_32424059/article/details/100612889

Leetcode 同时被 2 个专栏收录

821 篇文章

订阅专栏

Python

664 篇文章

订阅专栏

本文介绍了一种算法，用于计算给定字符串中只含单一字母的子串个数。通过两种不同思路实现：一是遍历所有子串并检查；二是优化后的数学方法，找出连续的单一字母子串，利用等差数列求和公式快速得出结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给你一个字符串 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(object):
    def countLetters(self, S):
        """
        :type S: str
        :rtype: int
        """
        res = 0
        for i in range(len(S)):
            for j in range(i, len(S)):
                t = S[i:j + 1]
                if t == t[0] * len(t):
                    res += 1
        return res

第二种思路：

按照数学的方法，先找出由单一字母构成的连续子串的长度N，

然后（N + 1） * N 就是这一个子串的答案的数量。

比如 "aaaba"，可以分成 "aaa" "b" "a"，

"aaa" 可以构成满足题目要求的子串 "aaa", "aa", "aa", "a", "a", "a", 答案数相当于从 1 + 2 + ... + N。

class Solution:
    def countLetters(self, s: str) -> int:
        # find substring with only one letter, whose len is k, then for this substring, res = 1 + 2+ ..k
        # TC: O(N)
        # SC: O(1)
        res = i = 0
        while i < len(s):
            j = i
            while j < len(s) and s[j] == s[i]:
                j += 1
            k = j - i
            res += k * (k + 1) // 2
            i = j
        return res