給你一個整數數組 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
用滑動窗口可計算出大于等于k個奇數數字的子數組數量,再減去大于等于k+1個奇數數字的子數組數量,即為優美子數組數量:
class Solution {
public:int numberOfSubarrays(vector<int>& nums, int k) {return getNum(nums, k) - getNum(nums, k + 1);}int getNum(vector<int> &nums, int target) {int left = 0;int oddNum = 0;int ret = 0;for (int i = 0; i < nums.size(); ++i) {oddNum += nums[i] & 1;while (oddNum >= target) {oddNum -= nums[left] & 1;++left;}ret += left;}return ret;}
};
如果nums大小為n,則此算法時間復雜度為O(n),空間復雜度為O(1)。