Untitled
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...