如何编写优质countVowels元音统计函数？求代码评审与优化

一、两种现有方案的问题分析 #

第一种方案（嵌套循环） #

• 性能冗余：两层嵌套循环带来O(n*m)的时间复杂度（n为单词长度，m为元音数组长度）。虽然元音只有5个，实际性能影响不大，但逻辑上存在无用操作——找到匹配元音后仍会遍历剩余元音，做不必要的比较。
• 可读性不足：嵌套结构比单层循环更难快速理解逻辑，增加了阅读和维护的成本。
• 变量命名不精准：用index作为计数变量不合适，index通常指代索引位置，这里应该用count或vowelCount这类语义明确的名称。

第二种方案（includes判断） #

• 性能仍有优化空间：vowels.includes()本质还是遍历数组查找元素，时间复杂度依旧是O(n*m)，和第一种方案底层逻辑类似，只是写法更简洁。
• 大小写兼容性缺失：仅处理小写元音，若输入包含大写字母（如"Apple"），会漏掉大写元音，不符合实际使用场景。
• 重复初始化问题：每次调用函数都会重新创建vowels数组，属于不必要的重复操作。
• 变量命名问题：同样存在index命名不准确的问题。

二、理想实现方案及原因 #

优化后的代码（迭代式） #

const VOWELS = new Set(['a', 'e', 'i', 'o', 'u', 'A', 'E', 'I', 'O', 'U']);

function countVowels(word) {
    let vowelCount = 0;
    for (const char of word) {
        if (VOWELS.has(char)) {
            vowelCount++;
        }
    }
    return vowelCount;
}

优化点及原因 #

1. 用Set存储元音：Set的has()方法时间复杂度为O(1)，整体时间复杂度降至O(n)，性能远优于数组查找，长单词场景下优势更明显。
2. 兼容大小写：直接包含大小写元音，覆盖所有常见输入场景；也可统一将输入转成小写（word.toLowerCase()）后判断，两种方式按需选择即可。
3. 常量外置：将元音集合定义为全局常量（大写命名标识常量），避免每次调用函数重复初始化，节省内存和初始化时间。
4. 语义化命名：vowelCount清晰表达变量用途，提升代码可读性。
5. for...of循环：相比传统for循环，写法更简洁，直接遍历字符，无需手动处理索引，逻辑更直观。

函数式风格替代方案 #

若偏好函数式编程，可采用更简洁的写法：

const VOWELS = new Set(['a', 'e', 'i', 'o', 'u', 'A', 'E', 'I', 'O', 'U']);

function countVowels(word) {
    return [...word].filter(char => VOWELS.has(char)).length;
}

逻辑：将字符串转为字符数组，用filter筛选出元音字符，最终返回筛选后数组的长度，代码简洁且可读性强。

内容的提问来源于stack exchange，提问作者kjs29