Skip to content

Latest commit

 

History

History
152 lines (117 loc) · 3.06 KB

File metadata and controls

152 lines (117 loc) · 3.06 KB

English Version

题目描述

给你一个整数数组 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

解法

Python3

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

Java

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;
    }
}

C++

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;
    }
};

Go

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)
}

TypeScript

...