Number of Zero-Filled Subarrays LeetCode solution in C++

1
class Solution {
2
 public:
3
  long long zeroFilledSubarray(vector<int> &nums) {
4
    long long count = 0;
5
    long long ans = 0;
6

7
    for (int i = 0; i < nums.size(); i++) {
8
      if (nums[i] == 0) {
9
        count++;
10
        ans += count;
11
      } else {
12
        count = 0;
13
      }
14
    }
15
    return ans;
16
  }
17
};