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

题目:

给你一个下标从 0 开始的字符串数组 words 以及一个二维整数数组 queries

每个查询 queries[i] = [l, r] 会要求我们统计在 words 中下标在 lr 范围内(包含 这两个值)并且以元音开头和结尾的字符串的数目。

返回一个整数数组,其中数组的第 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 <= 105
  • 1 <= words[i].length <= 40
  • words[i] 仅由小写英文字母组成
  • sum(words[i].length) <= 3 * 105
  • 1 <= queries.length <= 105
  • 0 <= queries[j][0] <= queries[j][1] < words.length

代码实现:

class Solution {
    public:
        vector<int> vowelStrings(vector<string>& words, vector<vector<int>>& queries) {
            // 前缀和数组处理
            // cnt[i] 表示 words数组中 前i个单词中的 元音字符串
            vector<int> cnt(words.size() + 1, 0);
            for(int i = 0; i < words.size(); ++i){
                cnt[i + 1] = cnt[i] + isValid(words[i]);
            }

            // 遍历求解
            vector<int> res;
            for(const auto& q : queries){
                // queries 数组为 [l, r] 左闭右闭,因此 为 cnt[l + 1] - cnt[r];
                res.emplace_back(cnt[q[1] + 1] - cnt[q[0]]);
            }
            return res;
        }

        bool isValid(const string& str){
            return valid(*str.begin()) && valid(*(--str.end()));
        }
        bool valid(char c){
            return !((c - 'a') && (c - 'e') && (c - 'i') && (c - 'o') && (c - 'u'));
        }
    };
posted @   DL1024  阅读(19)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 终于写完轮子一部分:tcp代理 了,记录一下
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 别再用vector<bool>了!Google高级工程师:这可能是STL最大的设计失误
· 单元测试从入门到精通
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
点击右上角即可分享
微信分享提示