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

题目

给你一个字符串 s ，返回 s 中 同构子字符串 的数目。由于答案可能很大，只需返回对 109 + 7 取余 后的结果。

同构字符串 的定义为：如果一个字符串中的所有字符都相同，那么该字符串就是同构字符串。

子字符串 是字符串中的一个连续字符序列。

思路

对于一连串相同的字串，例如“zzzzz” 他的同构子字符串的数目就是5 *（5 + 1 ） / 2，总共15个，该式子代表的是 1 + 2 + 3 + 4 + 5（分别代表1个z 2 个z … 5 个z）。

代码

class Solution {
public:
int countHomogenous(string s) {
int n = s.size();
long long ans = 0;
const int mod = 1e9 + 7;
for(int i = 0 , j = 0; i < n; i = j){
while(j < n && s[j] == s[i]){
j++;
}
long long cnt = j - i;
ans += cnt * (cnt + 1) / 2;
ans %= mod;
}
return ans;
}
};