LeetCode——1759. 统计同构子字符串的数目[Count Number of Homogenous Substrings][中等]——分析及代码[Java]
一、题目
给你一个字符串 s ,返回 s 中 同构子字符串 的数目。由于答案可能很大,只需返回对 10^9 + 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 由小写字符串组成
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/count-number-of-homogenous-substrings
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二、分析及代码
1. 直接计算
(1)思路
对一段长度为 l 的由相同字符构成的字符串,其同构子字符串的数量为 1 + 2 + … + l = l * (l + 1) / 2,直接计算即可。
因为计算结果可能超出 int 范围,需注意类型转换。
(2)代码
class Solution {
public int countHomogenous(String s) {
char[] c = s.toCharArray();
int n = c.length, m = 1000000007, i = 0, ans = 0;
while (i < n) {
int len = 1;
while (i + len < n && c[i + len] == c[i])
len++;
ans += ((len * (len + 1L)) % m) >> 1;
i += len;
}
return ans;
}
}
(3)结果
执行用时 :9 ms,在所有 Java 提交中击败了 100.00% 的用户;
内存消耗 :39 MB,在所有 Java 提交中击败了 100.00% 的用户。
三、其他
暂无。