Untitled
unknown
c_cpp
5 years ago
530 B
4
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...