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] 。

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/count-vowel-strings-in-ranges
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

 思路 :  前缀和. 对于每一次查询,统计区间内的个数. 

class Solution {
public:
    
    vector<int> vowelStrings(vector<string>& words, vector<vector<int>>& queries) {
        int n = words.size() ; 
        vector<int> ans ; 
        vector<int> w(26,0) ; 
        string ps = "aeiou" ; 
        for(char p :ps ){
            w[p-'a']++ ; 
        }
        vector<int> a(n) ;; 
        for(int i = 0 ; i<n ; i++ ){
            if(isVowel(words[i] ,w)){
                a[i] = 1 ; 
            }
        }
        vector<int> sum(n+1) ; 
        for(int i =1 ; i<=n ; i++ ){
            sum[i] = sum[i-1]+a[i-1] ; 
        }

        for(int i = 0 ; i<queries.size() ; i ++ ) {
            int l  = queries[i][0] ;
            int r = queries[i][1] ;  
            // cout<<l <<" " << r<< endl ; 
            int rt = sum[r+1] -sum[l] ;
            // cout<<<<endl; 
            ans.push_back(rt) ; 

        }
        return ans  ; 
    }
    // 判断是否是字符串是否符合要求
    bool isVowel(string s ,vector<int> w ){
        
        if(w[s[0] -'a'] >0  && w[s[s.size()-1] -'a'] >0)return true ;
        return false ; 
    }
};

猜你喜欢

转载自blog.csdn.net/qq_41661809/article/details/131007794