Question

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

English Version

题目描述

给你一个下标从 0 开始的字符串数组 words 和两个整数：left 和 right 。

如果字符串以元音字母开头并以元音字母结尾，那么该字符串就是一个 元音字符串 ，其中元音字母是 'a'、'e'、'i'、'o'、'u' 。

返回_ _words[i] 是元音字符串的数目，其中_ _i 在闭区间 [left, right] 内。

 

示例 1：

输入：words = ["are","amy","u"], left = 0, right = 2
输出：2
解释：
- "are" 是一个元音字符串，因为它以 'a' 开头并以 'e' 结尾。
- "amy" 不是元音字符串，因为它没有以元音字母结尾。
- "u" 是一个元音字符串，因为它以 'u' 开头并以 'u' 结尾。
在上述范围中的元音字符串数目为 2 。

示例 2：

输入：words = ["hey","aeo","mu","ooo","artro"], left = 1, right = 4
输出：3
解释：
- "aeo" 是一个元音字符串，因为它以 'a' 开头并以 'o' 结尾。
- "mu" 不是元音字符串，因为它没有以元音字母开头。
- "ooo" 是一个元音字符串，因为它以 'o' 开头并以 'o' 结尾。
- "artro" 是一个元音字符串，因为它以 'a' 开头并以 'o' 结尾。
在上述范围中的元音字符串数目为 3 。

 

提示：

• 1 <= words.length <= 1000
• 1 <= words[i].length <= 10
• words[i] 仅由小写英文字母组成
• 0 <= left <= right < words.length

解法

方法一：模拟

我们只需要遍历区间 $[left,.. right]$ 内的字符串，判断其是否以元音字母开头和结尾即可。若是，则答案加一。

遍历结束后，返回答案即可。

时间复杂度 $O(m)$，空间复杂度 $O(1)$。其中 $m = right - left + 1$。

class Solution:
  def vowelStrings(self, words: List[str], left: int, right: int) -> int:
    return sum(
      w[0] in 'aeiou' and w[-1] in 'aeiou' for w in words[left : right + 1]
    )

class Solution {
  public int vowelStrings(String[] words, int left, int right) {
    int ans = 0;
    for (int i = left; i <= right; ++i) {
      var w = words[i];
      if (check(w.charAt(0)) && check(w.charAt(w.length() - 1))) {
        ++ans;
      }
    }
    return ans;
  }

  private boolean check(char c) {
    return c == 'a' || c == 'e' || c == 'i' || c == 'o' || c == 'u';
  }
}

class Solution {
public:
  int vowelStrings(vector<string>& words, int left, int right) {
    auto check = [](char c) -> bool {
      return c == 'a' || c == 'e' || c == 'i' || c == 'o' || c == 'u';
    };
    int ans = 0;
    for (int i = left; i <= right; ++i) {
      auto w = words[i];
      ans += check(w[0]) && check(w[w.size() - 1]);
    }
    return ans;
  }
};

func vowelStrings(words []string, left int, right int) (ans int) {
  check := func(c byte) bool {
    return c == 'a' || c == 'e' || c == 'i' || c == 'o' || c == 'u'
  }
  for _, w := range words[left : right+1] {
    if check(w[0]) && check(w[len(w)-1]) {
      ans++
    }
  }
  return
}

function vowelStrings(words: string[], left: number, right: number): number {
  let ans = 0;
  const check: string[] = ['a', 'e', 'i', 'o', 'u'];
  for (let i = left; i <= right; ++i) {
    const w = words[i];
    if (check.includes(w[0]) && check.includes(w.at(-1))) {
      ++ans;
    }
  }
  return ans;
}

impl Solution {
  pub fn vowel_strings(words: Vec<String>, left: i32, right: i32) -> i32 {
    let check = |c: u8| -> bool {
      c == b'a' || c == b'e' || c == b'i' || c == b'o' || c == b'u'
    };

    let mut ans = 0;
    for i in left..=right {
      let w = words[i as usize].as_bytes();
      if check(w[0]) && check(w[w.len() - 1]) {
        ans += 1;
      }
    }

    ans
  }
}