leetcode 2348. 全 0 子数组的数目 中等

给你一个整数数组 nums ，返回全部为 0 的 子数组 数目。

子数组 是一个数组中一段连续非空元素组成的序列。

示例 1：

输入：nums = [1,3,0,0,2,0,0,4]
输出：6
解释：
子数组 [0] 出现了 4 次。
子数组 [0,0] 出现了 2 次。
不存在长度大于 2 的全 0 子数组，所以我们返回 6 。

示例 2：

输入：nums = [0,0,0,2,0,0]
输出：9
解释：
子数组 [0] 出现了 5 次。
子数组 [0,0] 出现了 3 次。
子数组 [0,0,0] 出现了 1 次。
不存在长度大于 3 的全 0 子数组，所以我们返回 9 。

示例 3：

输入：nums = [2,10,2019]
输出：0
解释：没有全 0 子数组，所以我们返回 0 。

提示：

• 1 <= nums.length <= 10^5
• -10^9 <= nums[i] <= 10^9

分析：遍历整个 nums 数组，一段连续的 0 能构成的全 0 子数组的个数为 (1+len)*len/2，其中 len 为连续的 0 的个数。

long long zeroFilledSubarray(int* nums, int numsSize) {
    long long ans=0,len=0;
    for(int i=0;i<numsSize;++i)
    {
        if(!nums[i])len++;
        else ans+=(1+len)*len/2LL,len=0;
    }
    ans+=(1+len)*len/2LL;
    return ans;
}