C++描述 LeetCode 5677. 统计同构子字符串的数目

  大家好,我叫亓官劼(qí guān jié )


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

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

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

示例 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 <= 105​
  • ​s​​ 由小写字符串组成

解题思路

遍历字符串,每次记录当前位置的值,向后遍历,如果相同,则每次cnt+k(如果当前是连续第k个,则可以分别多1,2,3 ……k个字母的同构,所以对于第k个连续字母,cnt+k)。

算法实现

class Solution {
public:
int countHomogenous(string s) {
int len = s.length();
int i = 0;
long long int cnt = 0;
while(i < len){
char temp = s[i];
cnt = (cnt + 1) % 1000000007;
i++;
int k = 1;
while(s[i] == temp){
k++;
cnt = (cnt + k) %1000000007;
i++;
}
}
return cnt;
}
};