要求
给你一个字符串 word ,返回 word 的所有子字符串中 元音的总数 ,元音是指 'a'、'e'、'i'、'o' 和 'u' 。
子字符串 是字符串中一个连续(非空)的字符序列。
注意:由于对 word 长度的限制比较宽松,答案可能超过有符号 32 位整数的范围。计算时需当心。
示例 1:
输入:word = "aba"
输出:6
解释:
所有子字符串是:"a"、"ab"、"aba"、"b"、"ba" 和 "a" 。
- "b" 中有 0 个元音
- "a"、"ab"、"ba" 和 "a" 每个都有 1 个元音
- "aba" 中有 2 个元音
因此,元音总数 = 0 + 1 + 1 + 1 + 1 + 2 = 6 。
示例 2:
输入:word = "abc"
输出:3
解释:
所有子字符串是:"a"、"ab"、"abc"、"b"、"bc" 和 "c" 。
- "a"、"ab" 和 "abc" 每个都有 1 个元音
- "b"、"bc" 和 "c" 每个都有 0 个元音
因此,元音总数 = 1 + 1 + 1 + 0 + 0 + 0 = 3 。
示例 3:
输入:word = "ltcd"
输出:0
解释:"ltcd" 的子字符串均不含元音。
示例 4:
输入:word = "noosabasboosa"
输出:237
解释:所有子字符串中共有 237 个元音。
提示:
- 1 <= word.length <= 105
- word 由小写英文字母组成
核心代码
class Solution:
def countVowels(self, word: str) -> int:
n = len(word)
res = 0
for i in range(n):
if word[i] in 'aeiou':
res += (i + 1) * (n - i)
return res
解题思路:使用排列组合的思想
1.题目是求所有的子串中,元音字母的个数
2.有2种数法:
(1)枚举每一个子串,然后数元音字母的个数
(2)枚举每一个元音字母(每一个index),数它的“贡献”次数(即可以出现在多少个子串中)
3.按照2(2)
如果word[i]是个元音字母。
子串必然是 word[L, ..., i, ..., R]
L有(i+1)种取法,R有(n-i)种取法。
====word[i]可以出现在 (i+1) * (n-i)个子串中
====word[i]贡献 (i+1) * (n-i)次