Untitled

 avatar
unknown
c_cpp
4 years ago
530 B
1
Indexable
#include <iostream>

using namespace std;

int main(){
    long long n;
    cin >> n;
    int a[n + 5];
    for(int i = 1; i <= n; i++) {
        cin >> a[i];
    }
    long long result = 0;
    a[0] = 0;
    a[n + 1] = 0;
    long long sz = 0;
    for(int i = 0; i <= n + 1; i++) {
        if(a[i] == 0) {
            result += sz * (sz + 1) / 2;
            sz = 0;
        }
        else {
            sz++;
        }
    }
    long long all_subarrays = n * (n + 1) / 2;
    cout << all_subarrays - result << endl;
return 0;
}
Editor is loading...