3306.js

/** ------------------------------------------------------------------------------
 * 
 * 3306. Count of Substrings Containing Every Vowel and K Consonants II
 * Topics: Sliding Window
 * https://leetcode.com/problems/count-of-substrings-containing-every-vowel-and-k-consonants-ii/description/?envType=daily-question&envId=2025-03-10
 * 
 ------------------------------------------------------------------------------ */
function countOfSubstrings(word, k) {
  return atMost(word, k) - atMost(word, k - 1)
}

function atMost(word, k) {
  let a = 0,
    e = 0,
    i = 0,
    o = 0,
    u = 0
  let count = 0
  for (let l = 0, r = 0; r < word.length; r++) {
    switch (word[r]) {
      case "a":
        a++
        break
      case "e":
        e++
        break
      case "i":
        i++
        break
      case "o":
        o++
        break
      case "u":
        u++
        break
      default:
        k--
    }

    while (a && e && i && o && u && k < 0) {
      switch (word[l++]) {
        case "a":
          a--
          break
        case "e":
          e--
          break
        case "i":
          i--
          break
        case "o":
          o--
          break
        case "u":
          u--
          break
        default:
          k++
      }
    }
    count += r - l + 1
  }
  return count
}

console.log(countOfSubstrings("abcabc", 3))