leetcode 1248.优美子数组 官答分析笔记

题目链接
题目:
给你一个整数数组 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

官答二:

class Solution {
    vector<int> cnt;
public:
    int numberOfSubarrays(vector<int>& nums, int k) {
        int n = (int)nums.size();
        cnt.resize(n + 1, 0);
        int odd = 0, ans = 0;
        cnt[0] = 1;
        for (int i = 0; i < n; ++i) {
            odd += nums[i] & 1;
            ans += odd >= k ? cnt[odd - k] : 0;
            cnt[odd] += 1;
        }
        return ans;
    }
};

分析:
官方题解下cnt数组一直没搞明白到底是怎么个记法,后来一步步调试才大致看懂。cnt数组的第i位元素a,表示的是父数组里以第i个奇数做优美子数组里首个奇数的优美子数组数量(且子数组里最后一位不是偶数时,这个情况要单独说明)
以以下示例举例:

输入:nums = [2,2,2,1,2,1,2,1,2], k = 2
输出:12

在开头遍历过程中cnt[0]不断增大,直到变成4才不再增加,再来看一下包含nums[3]这个奇数的有几个数组:
(1)[2,2,2,1,2,1]
(2)[2,2,1,2,1]
(3)[2,1,2,1]
(4)[1,2,1]
1.而当遍历到nums[4]时,ans=ans+cnt[0]=ans+4。
这里比较好理解,就是前面的这4个数组末尾加了个2;而此时cnt[0]不再变化,而是cnt[1]++,因为这个2相当于是第2个奇数做最优子数组开头奇数的情况了,就如同前面cnt[0]增大的情况,这里也是这么计算。因为第2个奇数和第3个奇数之间只有1个2,因此cnt[1]只增到2,ans的增加也是改为了ans=ans+cnt[1]=ans+2。

2.遍历到了到了第3个奇数,此时cnt[1]停止增加,cnt[2]开始增加,但在这个示例里其实跟它关系不大,为什么呢?因为并没有以第3个奇数做开头的优美子数组,cnt[2]的增加只是在为后续还可能出现的奇数做准备。只要不出现第4个奇数,cnt[2]就永远用不到。

真的想了很久,思路感觉和官方题解不一样,主要是官方题解着实看不懂啊。。。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值