【Leetcode 2062】 统计字符串中的元音子字符串 —— 双层循环,时间空间执行完胜

57 阅读2分钟

子字符串 是字符串中的一个连续(非空)的字符序列。

元音子字符串 是  由元音('a''e''i''o' 和 'u')组成的一个子字符串,且必须包含 全部五种 元音。

给你一个字符串 word ,统计并返回 word 中 元音子字符串的数目 。

示例 1:

输入: word = "aeiouu"
输出: 2
解释: 下面列出 word 中的元音子字符串(斜体加粗部分):
- "aeiouu"
- "aeiouu"

示例 2:

输入: word = "unicornarihan"
输出: 0
解释: word 中不含 5 种元音,所以也不会存在元音子字符串。

示例 3:

输入: word = "cuaieuouac"
输出: 7
解释: 下面列出 word 中的元音子字符串(斜体加粗部分):
- "cuaieuouac"
- "cuaieuouac"
- "cuaieuouac"
- "cuaieuouac"
- "cuaieuouac"
- "cuaieuouac"
- "cuaieuouac"

示例 4:

输入: word = "bbaeixoubb"
输出: 0
解释: 所有包含全部五种元音的子字符串都含有辅音,所以不存在元音子字符串。

双层循环,时间空间执行完胜

/*
https://leetcode.cn/u/cshappyeveryday/
执行用时:94 ms, 在所有 Typescript 提交中击败了100.00%的用户
内存消耗:52.35 MB, 在所有 Typescript 提交中击败了100.00%的用户
时间复杂度:O((n-4) * (n - i))
2024年8月26日 
*/
function countVowelSubstrings(word: string): number {
  const vowelSet = new Set<string>(["a", "e", "i", "o", "u"]);
  const n = word.length;
  const charSet = new Set<string>();
  let count = 0;
  for (let i = 0; i < n - 4; i++) {
    charSet.clear();
    for (let j = i; j < n; j++) {
      //不断扩充子串
      charSet.add(word[j]);
      //此处,如果 charSet 的长度为6,则证明有非元音字符,会在此步骤退出循环
      if (!isVowel(charSet)) break;
      //元音字符小于5,则不满足 5个元音字符。不可能大于5,Set会去重,
      if (charSet.size === 5) count++;
    }
  }
  //判断子串是否存在非元音,存在则表示新添加的字符中断了元音子串,该内层循环应该结束
  function isVowel(charSet: Set<string>) {
    for (const char of charSet) {
      if (!vowelSet.has(char)) return false;
    }
    return true;
  }
  return count;
}