2559. 统计范围内的元音字符串数 Medium
给你一个下标从 0 开始的字符串数组 words 以及一个二维整数数组 queries 。
每个查询 queries[i] = [li, ri] 会要求我们统计在 words 中下标在 li 到 ri 范围内（包含 这两个值）并且以元音开头和结尾的字符串的数目。
返回一个整数数组，其中数组的第 i 个元素对应第 i 个查询的答案。
注意：元音字母是 'a'、'e'、'i'、'o' 和 'u' 。
示例 1:
输入：words = ["aba","bcb","ece","aa","e"], queries = [[0,2],[1,4],[1,1]]
输出：[2,3,0]
解释：以元音开头和结尾的字符串是 "aba"、"ece"、"aa" 和 "e" 。
查询 [0,2] 结果为 2（字符串 "aba" 和 "ece"）。
查询 [1,4] 结果为 3（字符串 "ece"、"aa"、"e"）。
查询 [1,1] 结果为 0 。
返回结果 [2,3,0] 。
示例 2:
输入：words = ["a","e","i"], queries = [[0,2],[0,1],[2,2]]
输出：[3,2,1]
解释：每个字符串都满足这一条件，所以返回 [3,2,1] 。
解题思路
输入：一个字符串数组 words 和一个二维数组 queries，其中每个查询 [l, r] 表示要求统计 words 中下标从 l 到 r 范围内，以元音字母开头且结尾的字符串数量。
输出：返回一个数组，第 i 个元素表示第 i 个查询的结果。
本题属于一维前缀和问题。
我们可以先遍历 words 构建一个前缀和数组 pre_sum，其中 pre_sum[i] 表示下标在 [0, i-1] 范围内满足条件的字符串数量。
接着，在处理每个查询 [l, r] 时，利用前缀和快速计算出结果：
pre_sum[r + 1] - pre_sum[l] 即为下标区间 [l, r] 中满足条件的字符串数量
代码实现
from typing import List
class Solution:
def vowelStrings(self, words: List[str], queries: List[List[int]]) -> List[int]:
vowels = {'a', 'e', 'i', 'o', 'u'} # 用集合快速判断是否是元音
n = len(words)
# 构建前缀和数组，pre_sum[i] 表示 words[0:i] 中满足条件的单词个数
pre_sum = [0] * (n + 1)
for i in range(n):
# 判断当前单词首尾是否都是元音
if words[i][0] in vowels and words[i][-1] in vowels:
pre_sum[i + 1] = pre_sum[i] + 1
else:
pre_sum[i + 1] = pre_sum[i]
# 对每个查询区间 [l, r]，计算满足条件的单词个数
result = []
for l, r in queries:
result.append(pre_sum[r + 1] - pre_sum[l])
return result复杂度分析
时间复杂度：O(n)
空间复杂度：O(n)