title
给你一个字符串 s 和一个正整数 k 。
用 vowels 和 consonants 分别表示字符串中元音字母和辅音字母的数量。
如果某个字符串满足以下条件,则称其为 美丽字符串 :
vowels == consonants,即元音字母和辅音字母的数量相等。(vowels * consonants) % k == 0,即元音字母和辅音字母的数量的乘积能被k整除。
返回字符串 s 中 非空美丽子字符串 的数量。
子字符串是字符串中的一个连续字符序列。
英语中的 元音字母 为 'a'、'e'、'i'、'o' 和 'u' 。
英语中的 辅音字母 为除了元音字母之外的所有字母。
示例 1:
输入:s = "baeyh", k = 2
输出:2
解释:字符串 s 中有 2 个美丽子字符串。
- 子字符串 "baeyh",vowels = 2(["a","e"]),consonants = 2(["y","h"])。
可以看出字符串 "aeyh" 是美丽字符串,因为 vowels == consonants 且 vowels * consonants % k == 0 。
- 子字符串 "baeyh",vowels = 2(["a","e"]),consonants = 2(["b","y"])。
可以看出字符串 "baey" 是美丽字符串,因为 vowels == consonants 且 vowels * consonants % k == 0 。
可以证明字符串 s 中只有 2 个美丽子字符串。
示例 2:
输入:s = "abba", k = 1
输出:3
解释:字符串 s 中有 3 个美丽子字符串。
- 子字符串 "abba",vowels = 1(["a"]),consonants = 1(["b"])。
- 子字符串 "abba",vowels = 1(["a"]),consonants = 1(["b"])。
- 子字符串 "abba",vowels = 2(["a","a"]),consonants = 2(["b","b"])。
可以证明字符串 s 中只有 3 个美丽子字符串。
示例 3:
输入:s = "bcdf", k = 1
输出:0
解释:字符串 s 中没有美丽子字符串。
提示:
1 <= s.length <= 5 * 1041 <= k <= 1000s仅由小写英文字母组成。
solution
第一个条件是前缀和计数问题,key 是 前缀和。
但第二个条件需要我们转换下 子数组端点需要满足条件。对 i有要求,我们可以扩充前缀和的 key。
注意里面有个数学问题,已知L*L % 4k == 0, 则L 和 k 的关系。
class Solution:
def sqrt(self, n: int) -> int:
res = 1
i = 2
while i * i <= n:
i2 = i * i
while n % i2 == 0:
res *= i
n //= i2
if n % i == 0:
res *= i
n //= i
i += 1
if n > 1:
res *= n
return res
def beautifulSubstrings(self, s: str, k: int) -> int:
cnt = collections.Counter()
k = self.sqrt(4*k)
cnt[(k-1, 0)] = 1
presum = 0
ans = 0
for i,c in enumerate(s):
presum += 1 if c in "aeiou" else -1
p = (i % k, presum)
ans += cnt[p]
cnt[p] += 1
return ans