刷题使我快乐,满脸开心.jpg

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

题目

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

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

返回一个整数数组,其中数组的第 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

思路

感觉还是一个读题难度比较大的题目吧,
本来我的方法是再搞一个words等长的一维数组存words中每个字符串是否为符合条件的元音字符串,但是看了官方题解,感觉还是自己有点笨拙了,数学运用还是不够啊

代码

func vowelStrings(words []string, queries [][]int) []int {
    prefixSums := make([]int, len(words) + 1)
    for i := 0; i < len(words); i++ {
        prefixSums[i + 1] = prefixSums[i] + isVowelString(words[i])
    }
    ans := make([]int, len(queries))
    for i := 0; i < len(queries); i++ {
        ans[i] = prefixSums[queries[i][1] + 1] - prefixSums[queries[i][0]]
    }
    return ans
}

func isVowelString(word string) int {
    if isVowelLetter(word[0]) && isVowelLetter(word[len(word) - 1]) {
        return 1
    } else {
        return 0
    }
}

func isVowelLetter(c byte) bool {
    return c == 'a' || c == 'e' || c == 'i' || c == 'o' || c == 'u'
}