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

Posted 2021-12-07 心脏dance

题目链接：力扣

思路：很明显这是一个求贡献的题~~求每个元音字符的贡献，然后求和。首先大家肯定会想到暴力，以看数据量不行，那就想到求贡献~~

举个例子：abcdefg

e这个元音字符能做的贡献有哪儿些字符呢？

我是想到用两部分合起来的：

第一部分：abcde、bcde、cde、de、e、ef、efg 这一部分的贡献值是字符串的长度也就是7，怎么算呢？就是 len-i+i = len i表示当前字符的位置（从0开始）。这一部分的特点是e都在子字符串的边上，也就是没有跨过e的两边。

第二部分：abcdef、abcdefg、bcdef、bcdefg、cdef、cdefg、def、defg。这一部分的贡献值是e的两边的字符串的数目求乘积。怎么算呢？(len-i-1)*i i表示当前字符的位置（从0开始）。这一部分的特点是e都在子字符串的中间，跨过了e的两边。

注意：第二部分计算的时候会爆int（坑点）

上代码：

class Solution {
    fun countVowels(word: String): Long {
        var cnt = 0L
        val len = word.length
        for (i in word.indices) {
            if (word[i] == 'a' || word[i] == 'e' || word[i] == 'i' || word[i] == 'o' || word[i] == 'u') {
                cnt += (len - i - 1) * 1L * i + len - i + i
            }
        }
        return cnt
    }
}