Leetcode2063. 所有子字符串中的元音

2063. 所有子字符串中的元音

给你一个字符串 word ，返回 word 的所有子字符串中 元音的总数 ，元音是指 ‘a’、‘e’、‘i’、‘o’ 和 ‘u’ 。

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

注意：由于对 word 长度的限制比较宽松，答案可能超过有符号 32 位整数的范围。计算时需当心。

示例 1：

输入：word = “aba”
输出：6
解释：
所有子字符串是：“a”、“ab”、“aba”、“b”、“ba” 和 “a” 。

• “b” 中有 0 个元音
• “a”、“ab”、“ba” 和 “a” 每个都有 1 个元音
• “aba” 中有 2 个元音
  因此，元音总数 = 0 + 1 + 1 + 1 + 1 + 2 = 6 。

示例 2：

输入：word = “abc”
输出：3
解释：
所有子字符串是：“a”、“ab”、“abc”、“b”、“bc” 和 “c” 。

• “a”、“ab” 和 “abc” 每个都有 1 个元音
• “b”、“bc” 和 “c” 每个都有 0 个元音
  因此，元音总数 = 1 + 1 + 1 + 0 + 0 + 0 = 3 。

示例 3：

输入：word = “ltcd”
输出：0
解释：“ltcd” 的子字符串均不含元音。

示例 4：

输入：word = “noosabasboosa”
输出：237
解释：所有子字符串中共有 237 个元音。

提示：

1 <= word.length <= 105
word 由小写英文字母组成

题解

• 二维dp，超时：37 / 51 个通过测试用例

#include <iostream>
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
class Solution {
public:
    bool isVowel(char ch)
    {
        if (ch == 'a' || ch == 'e' || ch == 'i' || ch == 'o' || ch == 'u') {
            return true;
        }
        return false;
    }

    long long countVowels(string word) {
        int n = word.size();
        // dp[i][j]表示以第i个字符结尾长度为j的子串包含元音的个数
        vector<vector<ll>> dp(n + 1, vector<ll>(n + 1, 0));
        ll res = 0;
        for (int i = 1; i <= n; ++i) {
            for (int j = 1; j <= i; ++j) {
                dp[i][j] = dp[i - 1][j - 1] + (isVowel(word[i - 1]) ? 1 : 0);
                res += dp[i][j];
            }
        }
        return res;
    }
};

• 优化为一维dp，通过

#include <iostream>
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
class Solution {
public:
    bool isVowel(char ch)
    {
        if (ch == 'a' || ch == 'e' || ch == 'i' || ch == 'o' || ch == 'u') {
            return true;
        }
        return false;
    }

    long long countVowels(string word) {
        ll dp = 0;
        ll res = 0;
        for (int i = 1; i <= word.size(); ++i) {
            dp += (isVowel(word[i - 1]) ? i : 0); // 每增加一个元音可以增加i个包含元音的子串
            res += dp;
        }
        return res;
    }
};