文章目錄

• • 1. 題目
  • 2. 解題
  • • 2.1 記錄奇數出現的pos
    • 2.2 前綴和

1. 題目

給你一個整數數組 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

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/count-number-of-nice-subarrays
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

2. 解題

• 類似題目：LeetCode 560. 和為K的子數組（前綴和差分）

2.1 記錄奇數出現的pos

• 找到k個奇數左右的偶數個數，相乘就是方案數

class Solution { public:int numberOfSubarrays(vector<int>& nums, int k) {int i, cnt = 0, n = nums.size(), count = 0;vector<int> oddPos(n+2);for(i = 0; i < n; ++i){if(nums[i] & 1)//奇數oddPos[++cnt] = i;}oddPos[0] = -1, oddPos[++cnt] = n;//邊界，假設兩邊有0個偶數for(i = 1; i+k <= cnt; ++i)count += (oddPos[i]-oddPos[i-1])*(oddPos[i+k]-oddPos[i+k-1]);return count;} };

372 ms 66.1 MB

2.2 前綴和

class Solution { public:int numberOfSubarrays(vector<int>& nums, int k) {int i, oddcnt = 0, n = nums.size(), count = 0;vector<int> preOddCnt(n+1,0);preOddCnt[0] = 1;//邊界for(i = 0; i < n; ++i){oddcnt += (nums[i]&1);//奇數多少個了if(oddcnt >= k)//奇數夠個數了count += preOddCnt[oddcnt-k];//以i結束的長度為k個奇數的數組個數 preOddCnt[oddcnt-k]preOddCnt[oddcnt] += 1;//這么多個奇數的數組 +1 個}return count;} };

368 ms 65.8 MB

總結

以上是生活随笔為你收集整理的LeetCode 1248. 统计「优美子数组」（要复习）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。