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