LeetCode 2559. 统计范围内的元音字符串数

2559. 统计范围内的元音字符串数

给你一个下标从 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] 。

提示：

• 1 <= words.length <= 10^5
• 1 <= words[i].length <= 40
• words[i] 仅由小写英文字母组成
• sum(words[i].length) <= 3 * 10^5
• 1 <= queries.length <= 10^5
• 0 <= queries[j][0] <= queries[j][1] < words.length

提示 1

Precompute the prefix sum of strings that start and end with vowels.

提示 2

Use unordered_set to store vowels.

提示 3

Check if the first and last characters of the string are present in the vowels set.

提示 4

Subtract prefix sum for range [l-1, r] to find the number of strings starting and ending with vowels.

解法：前缀和 

class Solution {
    public int[] vowelStrings(String[] words, int[][] queries) {
        String vowel = "aeiou";
        int n = words.length;
        int[] pre = new int[n + 1];
        int[] ans = new int[queries.length];
        for (int i = 1; i <= n; i++) {
            String s = words[i - 1];
            pre[i] = ( vowel.indexOf(s.charAt(0)) != -1 && vowel.indexOf(s.charAt(s.length() - 1)) != -1 ) ? pre[i - 1] + 1 : pre[i - 1];
        }
        for (int i = 0; i < queries.length; i++) {
            ans[i] = pre[queries[i][1] + 1] - pre[queries[i][0]];
        }
        return ans;
    }
}

复杂度分析 

• 时间复杂度：O(N)，N 是数组words 和 数组queries 的长度 中的较大值。
• 空间复杂度：O(N)，N 是数组words 和 数组queries 的长度 中的较大值。