Leetcode 统计只含单一字母的子串

给你一个字符串 S，返回只含 单一字母 的子串个数。

示例 1：

输入： "aaaba"
输出： 8
解释： 
只含单一字母的子串分别是 "aaa"， "aa"， "a"， "b"。
"aaa" 出现 1 次。
"aa" 出现 2 次。
"a" 出现 4 次。
"b" 出现 1 次。
所以答案是 1 + 2 + 4 + 1 = 8。

示例 2:

输入： "aaaaaaaaaa"
输出： 55

 

这道题目从O(N^3)复杂度优化道O(N^2)复杂度Hash，再想到优化道O(N), 优化到O(N)开始想的是双指针，最后用找规律的方法解决。

class Solution {
public:
    int countLetters(string S) {
        char a = S[0];
        int res=0, temp = 0;
        for(auto c:S){
            if(a==c){
                temp++;
                res+=temp;
            }
            else {
                temp=1;
                res++;
                a = c;
            }
        }
        return res;
    }
};