leetcode 3305. 元音辅音字符串计数 I 中等

给你一个字符串 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 <= 250
• word 仅由小写英文字母组成。
• 0 <= k <= word.length - 5

分析：由于字符串长度不超过250，暴力法也可以解决，这里使用滑动窗口法。从左向右遍历字符串，直到满足条件。再从当前窗口的左端点向右移动，检查有多少个符合要求的子串。之后继续遍历，如果符合条件，则加上子串数量，同时继续检查有多少符合要求的子串；如果不符合条件，将子串数量重置。

注意遍历完成后，还要再检查一次当前的窗口是否符合条件。

int judge(char c)
{if(c=='a')return 0;if(c=='e')return 1;if(c=='i')return 2;if(c=='o')return 3;if(c=='u')return 4;return 5;
}
int countOfSubstrings(char* word, int k) {int ans=0,l=0,cnt=5,len=strlen(word),power=0;int sum[6]={0};for(int i=0;i<len;++i){int index=judge(word[i]);if(index<=4){if(sum[index]==0)cnt--;sum[index]++;}else if(index==5){if(sum[index]==k){ans+=power;for(;l<=i;++l){int pos=judge(word[l]);if(pos<=4){sum[pos]--;if(sum[pos]==0)cnt++;}else if(pos==5){sum[pos]--;if(sum[pos]<=k)break;}}power=0;if(sum[5]<k)sum[index]++,l++;}else sum[index]++;}if(sum[5]==k&&cnt==0){int temp[6]={0};for(int j=0;j<6;++j)temp[j]=sum[j];for(int j=l;j<i;++j){power++;int pos=judge(word[j]);temp[pos]--;if((pos<=4&&temp[pos]==0)||temp[5]<k)break;}}//printf("i=%d l=%d cnt=%d ans=%d k=%d pow=%d\n",i,l,cnt,ans,sum[5],power);}ans+=power;return ans;
}