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

Problem#

给你一个由 n 个字符串组成的字符串数组 words，以及一个左闭右开区间 [left, right)。

请你统计并返回在该区间内，下标 i 满足 left <= i < right 的字符串中，那些 首尾字符都是元音字母 的字符串的数目。

元音字母：a, e, i, o, u（不区分大小写）。

https://leetcode.cn/problems/count-vowel-substrings-of-a-string/description/

示例 1：

输入：words = [“are”,“amy”,“u”], left = 0, right = 2 输出：2 解释：在 [0, 2) 范围内：

“are” 首尾都是元音

“amy” 首不是元音，尾是元音

“u” 首尾都是元音因此返回 2。

示例 2：

输入：words = [“hey”,“aeo”,“mu”,“ooo”,“artro”], left = 1, right = 4 输出：3 解释：在 [1, 4) 范围内：

“aeo” 首尾都是元音

“mu” 首尾都不是元音

“ooo” 首尾都是元音因此返回 3。

约束：

1 <= words.length <= 1000
1 <= words[i].length <= 10
0 <= left < right <= words.length

思路#

遍历区间内的每个字符串，检查首尾字符是否在元音集合中。

1
class Solution:
2
    def vowelStrings(self, words: list[str], left: int, right: int) -> int:
3
        vowels = set('aeiou')
4
        count = 0
5
        for i in range(left, right):
6
            w = words[i]
7
            if w[0] in vowels and w[-1] in vowels:
8
                count += 1
9
        return count

1
import unittest
2

3

4
class TestVowelStrings(unittest.TestCase):
5
    def setUp(self):
6
        self.solution = Solution()
7

8
    def test_example1(self):
9
        self.assertEqual(self.solution.vowelStrings(["are", "amy", "u"], 0, 2), 2)
10

11
    def test_example2(self):
12
        self.assertEqual(self.solution.vowelStrings(["hey", "aeo", "mu", "ooo", "artro"], 1, 4), 2)
13

14
    def test_none(self):
15
        self.assertEqual(self.solution.vowelStrings(["bc", "df"], 0, 2), 0)
16

17

18
if __name__ == '__main__':
19
    unittest.main()