【前缀和】3306. 元音辅音字符串计数 II|2200

最新推荐文章于 2025-11-20 16:05:01 发布

原创最新推荐文章于 2025-11-20 16:05:01 发布 · 1.8k 阅读

28 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #算法 #力扣 #前缀和 #元音 #辅音 #计数

困难算法题专栏收录该内容

630 篇文章

订阅专栏

本文涉及的基础知识点

C++算法：前缀和、前缀乘积、前缀异或的原理、源码及测试用例包括课程视频

LeetCode3306. 元音辅音字符串计数 II

给你一个字符串 word 和一个非负整数 k。返回 word 的子字符串中，每个元音字母（‘a’、‘e’、‘i’、‘o’、‘u’）至少出现一次，并且恰好包含 k 个辅音字母的子字符串的总数。
示例 1：
输入：word = “aeioqq”, k = 1
输出：0
解释：
不存在包含所有元音字母的子字符串。
示例 2：
输入：word = “aeiou”, k = 0
输出：1
解释：
唯一一个包含所有元音字母且不含辅音字母的子字符串是 word[0…4]，即 “aeiou”。
示例 3：
输入：word = “ieaouqqieaouqq”, k = 1
输出：3
解释：
包含所有元音字母并且恰好含有一个辅音字母的子字符串有：
word[0…5]，即 “ieaouq”。
word[6…11]，即 “qieaou”。
word[7…12]，即 “ieaouq”。
提示：
5 <= word.length <= 2 * 10⁵
word 仅由小写英文字母组成。
0 <= k <= word.length - 5

前缀和

v[0…4]记录 word[0…i]之间为a o e i u的最大下标。如果不存在为，-1。
inx按顺序记录所有辅音字母的下标。
如果words[j…i]符合条件，则：
j > inx倒数第k+1个，j < inx倒数第k-1个 j <= min(v)

代码

核心代码

class Solution {
		public:
			long long countOfSubstrings(string word, const int K) {
				string strs = "aoeiu";
				vector<int> v(5, -1);
				vector<int> inxs;
				long long ans = 0;
				for (int i = 0; i < word.size();i++ ) {
					const auto& ch = word[i];
					int type = -1;
					for (int j = 0; j < strs.length(); j++) {
						if (ch == strs[j]) { type = j ; }
					}
					if (-1 == type) { inxs.emplace_back(i); }
					else { v[type] = i; }
					const int iMax1 = *min_element(v.begin(), v.end())+1;
					const int k = inxs.size();		
					if (k < K) { continue; }
					const int i2 = k - K ;
					const int iMax2 = (i2>=k) ? (i+1):( inxs[i2]+1);
					const int iMax = min(iMax1,iMax2 );
					const int i1 = k - K - 1;
					const int iMin = (i1 < 0) ? -1 : inxs[i1];
					const int cur = iMax - iMin - 1;
					ans += max(0, cur);
				}
				return ans;
			}
		};

单元测试

class Solution {
		public:
			long long countOfSubstrings(string word, const int K) {
				string strs = "aoeiu";
				vector<int> v(5, -1);
				vector<int> inxs;
				long long ans = 0;
				for (int i = 0; i < word.size();i++ ) {
					const auto& ch = word[i];
					int type = -1;
					for (int j = 0; j < strs.length(); j++) {
						if (ch == strs[j]) { type = j ; }
					}
					if (-1 == type) { inxs.emplace_back(i); }
					else { v[type] = i; }
					const int iMax1 = *min_element(v.begin(), v.end())+1;
					const int k = inxs.size();		
					if (k < K) { continue; }
					const int i2 = k - K ;
					const int iMax2 = (i2>=k) ? (i+1):( inxs[i2]+1);
					const int iMax = min(iMax1,iMax2 );
					const int i1 = k - K - 1;
					const int iMin = (i1 < 0) ? -1 : inxs[i1];
					const int cur = iMax - iMin - 1;
					ans += max(0, cur);
				}
				return ans;
			}
		};

单元测试

	string word;
		int k;
		TEST_METHOD(TestMethod11)
		{
			word = "aeioqq", k = 1;
			auto res = Solution().countOfSubstrings(word, k);
			AssertEx(0LL, res);
		}
		TEST_METHOD(TestMethod12)
		{
			word = "aeiou", k = 0;
			auto res = Solution().countOfSubstrings(word, k);
			AssertEx(1LL, res);
		}
		TEST_METHOD(TestMethod13)
		{
			word = "ieaouqqieaouqq", k = 1;
			auto res = Solution().countOfSubstrings(word, k);
			AssertEx(3LL, res);
		}

扩展阅读

我想对大家说的话
工作中遇到的问题，可以按类别查阅鄙人的算法文章，请点击《算法与数据汇总》。
学习算法：按章节学习《喜缺全书算法册》，大量的题目和测试用例，打包下载。重视操作
有效学习：明确的目标及时的反馈拉伸区（难度合适）专注
闻缺陷则喜(喜缺)是一个美好的愿望，早发现问题，早修改问题，给老板节约钱。
子墨子言之：事无终始，无务多业。也就是我们常说的专业的人做专业的事。
如果程序是一条龙，那算法就是他的是睛
失败+反思=成功成功+反思=成功