0 から始まる添字付きの文字列配列ワードと 2 次元の整数配列クエリが与えられます。
各クエリ queries[i] = [li, ri] は、母音で始まり母音で終わる、li から ri (両端を含む) の範囲の添え字を持つ単語内の文字列の数を数えるように求めます。
配列の i 番目の要素が i 番目のクエリに対する答えに対応する整数の配列を返します。
注: 母音は「a」、「e」、「i」、「o」、「u」です。
例 1:
入力: 単語 = ["aba","bcb","ece","aa","e"]、クエリ = [[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:
入力: 単語 = ["a","e","i"]、クエリ = [[0,2],[0,1],[2,2]] 出力: [3,2,1]
説明
:すべての文字列がこの条件を満たすため、[3,2,1] が返されます。
出典: LeetCode
リンク: https://leetcode.cn/problems/count-vowel-strings-in-ranges
著作権は LeetCode に属します。営利目的での転載の場合は正式な許諾を得た方へ、非営利での転載の場合は出典を明記の上、ご連絡ください。
アイデア: プレフィックスの合計 各クエリについて、間隔内の数値をカウントします。
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 ;
}
};