美文网首页
【算法题】2559. 统计范围内的元音字符串数

【算法题】2559. 统计范围内的元音字符串数

作者: 程序员小2 | 来源:发表于2023-04-11 07:53 被阅读0次

    题目:

    给你一个下标从 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) {
            // 构造元音哈希集合
            String s = "aeiou";
            HashSet<Character> cs = new HashSet<>();
            for (char c : s.toCharArray()) {
                cs.add(c);
            }
    
            // 构造前缀和数组
            int len = words.length;
            int[] prefixAndArr = new int[len];
            for (int i = 0; i < len; i++) {
                String word = words[i];
                if (isMatch(word, cs)) {
                    // 首尾皆是元音
                    if (i == 0) {
                        prefixAndArr[i] = 1;
                    } else {
                        prefixAndArr[i] += prefixAndArr[i - 1] + 1;
                    }
                } else {
                    if (i > 0) {
                        // 取上一次值
                        prefixAndArr[i] = prefixAndArr[i - 1] + 0;
                    }
                }
            }
    
            len = queries.length;
            int[] rs = new int[len];
            for (int i = 0; i < len; i++) {
                if (queries[i][0] == 0) {
                    // 直接取前缀和
                    rs[i] = prefixAndArr[queries[i][1]];
                } else {
                    // 求前缀和的差
                    // 注意起始索引多减一,相当于求两段的变化:(end - start) + (start - (start - 1))
                    rs[i] = prefixAndArr[queries[i][1]] - prefixAndArr[queries[i][0] - 1];
                }
            }
            return rs;
        }
    
        // 判断首尾是否皆是元音
        private boolean isMatch(String word, HashSet<Character> cs) {
            if (cs.contains(word.charAt(0)) && cs.contains(word.charAt(word.length() - 1))) {
                return true;
            } else {
                return false;
            }
        }
    }
    
    

    相关文章

      网友评论

          本文标题:【算法题】2559. 统计范围内的元音字符串数

          本文链接:https://www.haomeiwen.com/subject/tkcgddtx.html