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 <= 105^55
-109^99 <= nums[i] <= 109^99
分组循环,找出全0的一组,并计算子数组数量:
class Solution {
public:long long zeroFilledSubarray(vector<int>& nums) {int n = nums.size();long long ans = 0;int i = 0;while (i < n) {if (nums[i] != 0) {++i;continue;}int start = i;while (i < n && nums[i] == 0) {++i;}int len = i - start;ans += (long long)len * (len + 1) / 2;}return ans;}
};
如果nums的长度为n,则此算法时间复杂度为O(n),空间复杂度为O(1)。