【从零开始的LeetCode-算法】2559. 统计范围内的元音字符串数

九圣残炎2024-11-14 10:30

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

我的解答

java 复制代码

class Solution {
    public int[] vowelStrings(String[] words, int[][] queries) {
        int[] aeiou = new int[words.length];
        aeiou[0] = checkAeiou(words[0]);
        // 统计前i个符合条件的字符串数
        for(int i = 1; i < words.length; i++){
            aeiou[i] = aeiou[i - 1] +  checkAeiou(words[i]);
        }

        int[] res = new int[queries.length];
        for(int i = 0; i < queries.length; i++){
            // 用后区间的总数减去前区间的总数，即得当前区间范围（不包含前区间）的元音字符串总数
            // 补上前区间对应的字符串即范围内的元音字符串数
            res[i] = aeiou[queries[i][1]] - aeiou[queries[i][0]] + checkAeiou(words[queries[i][0]]);
        }
        return res;
    }

    private int checkAeiou(String str){
        char start = str.charAt(0),end = str.charAt(str.length() - 1);
        if((start == 'a' || start == 'e' ||  start == 'i'  || start == 'o'  || start == 'u')
                && (end == 'a' || end == 'e' ||  end == 'i'  || end == 'o'  || end == 'u')) {
            return 1;
        }else{
            return 0;
        }
    }
}