[Swift]LeetCode1248. 统计「优美子数组」| Count Number of Nice Subarrays

★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★
➤微信公众号：山青咏芝（let_us_code）
➤博主域名：https://www.zengqiang.org
➤GitHub地址：https://github.com/strengthen/LeetCode
➤原文地址：
➤如果链接不是山青咏芝的博客园地址，则可能是爬取作者的文章。
➤原文已修改更新！强烈建议点击原文地址阅读！支持作者！支持原创！
★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★

Given an array of integers nums and an integer k. A subarray is called nice if there are k odd numbers on it.

Return the number of nice sub-arrays.

Example 1:

Input: nums = [1,1,2,1,1], k = 3
Output: 2
Explanation: The only sub-arrays with 3 odd numbers are [1,1,2,1] and [1,2,1,1].
Example 2:

Input: nums = [2,4,6], k = 1
Output: 0
Explanation: There is no odd numbers in the array.
Example 3:

Input: nums = [2,2,2,1,2,2,1,2,2,2], k = 2
Output: 16

Constraints:

1 <= nums.length <= 50000
1 <= nums[i] <= 10^5
1 <= k <= nums.length

---

给你一个整数数组 nums 和一个整数 k。

如果某个 连续 子数组中恰好有 k 个奇数数字，我们就认为这个子数组是「优美子数组」。

请返回这个数组中「优美子数组」的数目。

示例 1：

输入：nums = [1,1,2,1,1], k = 3
输出：2
解释：包含 3 个奇数的子数组是 [1,1,2,1] 和 [1,2,1,1] 。
示例 2：

输入：nums = [2,4,6], k = 1
输出：0
解释：数列中不包含任何奇数，所以不存在优美子数组。
示例 3：

输入：nums = [2,2,2,1,2,2,1,2,2,2], k = 2
输出：16

提示：

1 <= nums.length <= 50000
1 <= nums[i] <= 10^5
1 <= k <= nums.length

---

1056ms

class Solution {
    func numberOfSubarrays(_ nums: [Int], _ k: Int) -> Int {
        var oddIndex = [Int]()
        var res = 0
        var p = 0
        for i in 0 ..< nums.count where nums[i] % 2 != 0 {
            if oddIndex.count >= k {
                res += (oddIndex[p] - (p > 0 ? oddIndex[p - 1] : -1)) * (i - oddIndex.last!)
                p += 1
            }
            oddIndex.append(i)
        }
        
        if oddIndex.count >= k {
            res += (oddIndex[p] - (p > 0 ? oddIndex[p - 1] : -1)) * (nums.count - oddIndex.last!)   
        }
        
        return res
    }
}

---

1068ms

class Solution {
    func numberOfSubarrays(_ nums: [Int], _ k: Int) -> Int {
        return atMostK(nums, k) - atMostK(nums, k - 1)
    }
    
    private func atMostK(_ nums: [Int], _ count: Int) -> Int {
        var count = count
        var res = 0
        var i = 0
        for j in 0..<nums.count {
            if nums[j] % 2 == 1 {
                count -= 1
            }
            
            while count < 0 {
                if nums[i] % 2 == 1 {
                    count += 1
                }
                i += 1
            }
            res += j - i + 1
        }
        return res
    }
}

---

1076ms

class Solution {
    func numberOfSubarrays(_ nums: [Int], _ k: Int) -> Int {
        if nums.count == 0 {
            return 0
        }
        var oddPosition : [Int] = []
        for i in 0...nums.count - 1 {
            if nums[i] % 2 == 1 {
                oddPosition.append(i)
            }
        }
        if oddPosition.count < k {
            return 0
        }
        oddPosition.insert(-1,at:0)
        oddPosition.append(nums.count)
        
        var result = 0
        for  i in 1...oddPosition.count - k - 1 {
            result += (oddPosition[i] - oddPosition[i-1])*(oddPosition[i+k] - oddPosition[i+k-1])
        }
        return result
    }
}