3305. 元音辅音字符串计数

问:某班有 10 个人至少 20 岁,3 个人至少 21 岁,那么恰好 20 岁的人有多少个?

答:「至少 20 岁」可以分成「恰好 20 岁」和「至少 21 岁」,所以「至少 20 岁」的人数减去「至少 21 岁」的人数,就是「恰好 20 岁」的人数,即 10−3=7。

根据这个思路(容斥原理),本题等价于如下两个问题:

处理一个恰好的问题,可以转化为两个至少的问题

每个元音字母至少出现一次,并且至少包含k个辅音字母的子串个数。记作f(k)
每个元音字母至少出现一次,并且至少包含k+1个辅音字母的子串个数。记作f(k+1)
2者相减,所表达的含义就是恰好包含k个辅音字母了,所以答案为f(k+1)-f(k)

本题中,子串越长,包含的元音越多

代码如何构思:

首先关于至少的问题,在滑动窗口的过程中,枚举子串的右端点,把右端点的字母放到窗口里面,为了判断出两个至少的条件(元音字母至少出现一次,至少包含k个辅音字母),得用两个东西来维护。

对于元音来说,每个元音的出现次数是不一样的,所以得用一个哈希表或者数组来维护每个元音的出现次数,辅音只要求总数是>=k,所以只需要一个变量来维护

当右端点进入窗口之后,根据右端点是元音还是辅音,来增加cnt1或者cnt2的值,当窗口的子串满足要求的时候。也就是当右端点固定,右端点满足要求的时候,要求出的是有多少个子串的左端点是满足要求的

怎么操作?当满足条件的时候,收缩左端点,左端点向右移动(左端点更新过程中,cnt1和cnt2的更新与上文相同)

当收缩结束时,left是什么?循环结束的时候从left到right是满足条件还是不满足条件?是不满足条件,才退出循环。但是退出循环的前一刻left-1是满足要求的,同理,前面的更长的子串是满足要求的

class Solution:
    def f(self,word: str, k: int) -> int:
        cnt1 = defaultdict(int)
        cnt2 = 0
        ans = left = 0
        for i in word:
            if i in 'aeiou':
                cnt1[i] += 1
            else:
                cnt2 += 1
            while len(cnt1) == 5 and cnt2 >= k:
                cur = word[left]
                if cur in 'aeiou':
                    cnt1[cur] -= 1
                    if cnt1[cur] == 0:
                        del cnt1[cur]
                else:
                    cnt2 -= 1
                left += 1
            ans += left 
        return ans
    def countOfSubstrings(self, word: str, k: int) -> int:
        return self.f(word,k) - self.f(word,k+1)
        

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值