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

 

给你一个字符串 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. 1 <= S.length <= 1000
2. 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(object):
    def countLetters(self, S):
        """
        :type S: str
        :rtype: int
        """
        i = 0
        pre = S[0]
        pre_start = 0
        res = 0
        while i < len(S):
            if S[i] != pre:
                n = i - pre_start
                res += n * (n + 1) // 2
                pre = S[i]
                pre_start = i
            i += 1
        n = i - pre_start
        res += n * (n + 1) // 2    
        return res