给你一个整数数组 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 <= 105
-109 <= nums[i] <= 109
class Solution:
def zeroFilledSubarray(self, nums: List[int]) -> int:
ans = 0
cnt = 0
for v in nums:
if v == 0:
cnt += 1
else:
ans += (1 + cnt) * cnt // 2
cnt = 0
ans += (1 + cnt) * cnt // 2
return ans
class Solution {
public long zeroFilledSubarray(int[] nums) {
long ans = 0;
int cnt = 0;
for (int v : nums) {
if (v == 0) {
++cnt;
} else {
ans += (long) (1 + cnt) * cnt / 2;
cnt = 0;
}
}
ans += (long) (1 + cnt) * cnt / 2;
return ans;
}
}
class Solution {
public:
long long zeroFilledSubarray(vector<int>& nums) {
long long ans = 0;
int cnt = 0;
for (int v : nums) {
if (v == 0)
++cnt;
else {
ans += 1ll * (1 + cnt) * cnt / 2;
cnt = 0;
}
}
ans += 1ll * (1 + cnt) * cnt / 2;
return ans;
}
};
func zeroFilledSubarray(nums []int) int64 {
ans := 0
cnt := 0
for _, v := range nums {
if v == 0 {
cnt++
} else {
ans += (1 + cnt) * cnt / 2
cnt = 0
}
}
ans += (1 + cnt) * cnt / 2
return int64(ans)
}