leecode.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

思路分析

• 对于一个长度为n的连续相同的字符串来说，其可以构成的长度为n+(n - 1)+…+1=(n+1)*n/2.
• 但是这道题不能用乘法做，因为数据范围可能爆int。n<=1e5.即使按照取模的分配律，结果也是会爆的。
• 所以只能按照加法来做

代码

const int mod = 1e9 + 7;
class Solution {
public:
    int countHomogenous(string s) {
        int n = s.size(), ans = 0, l = 0, r = 0;
        while(r < n){
            while(r < n && s[r] == s[l]){
                r++;
                ans = (ans + r - l) % mod;
            } 
            l = r;
        }
        while(l < n - 1){
            ans = (ans + n - l) % mod;
            l++;
        }
        return ans;
    }
};