LeetCode 1759. 统计同构子字符串的数目

示例 1：
输入：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

示例 2：
输入：s = "xy"
输出：2
解释：同构子字符串是 "x" 和 "y" 。

示例 3：
输入：s = "zzzzz"
输出：15
 
提示：
1 <= s.length <= 10^5
s 由小写字符串组成

class Solution {
public:
    int countHomogenous(string s) {
        int ans = 0, n = 0, num = 0, mod = 1e9+7;
        char prev = s[0];
        for(int i = 0; i < s.size(); ++i)
        {
            if(s[i] == prev)
            {
                n++;
            }
            else if(s[i] != prev)
            {
                prev = s[i];
                num = (1LL*(1+n)*n/2)%mod;
                n = 1;
                ans = (ans + num)%mod;
            }
        }
        num = (1LL*(1+n)*n/2)%mod;
        ans = (ans + num)%mod;
        return ans;
    }
};