1759. 统计同构子字符串的数目
难度中等45
给你一个字符串 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
由小写字符串组成
双指针(字符串计数)
class Solution {
private static final int MOD = (int)1e9+7;
public int countHomogenous(String s) {
int res = 0;
int left = 0, right = 0;
while(right < s.length()){
while(right < s.length() && s.charAt(left) == s.charAt(right)) right++;
res += count(right-left) % MOD;
left = right;
}
return res;
}
public int count(int n){
int res = 0;
while(n > 0){
res += n;
res %= MOD;
n--;
}
return res;
}
}
双指针
class Solution {
private static final int MOD = (int)1e9+7;
public int countHomogenous(String s) {
int res = 0;
int left = 0, right = 0;
while(right < s.length()){
if(s.charAt(right) != s.charAt(left)){
left = right;
}
res = (res + (right-left+1)) % MOD;
right++;
}
return res;
}
}