Submission #526697

#	Time	Username	Problem	Language	Result	Execution time	Memory
526697		joelau	Izbori (COCI22_izbori)	C++14	15 / 110	70 ms	14416 KiB

Main

#include <bits/stdc++.h>
using namespace std;

long long N,A[200005];
map<long long,long long> mp;

int main() {
    ios_base::sync_with_stdio(false); cin.tie(NULL);
    cin >> N;
    for (long long i = 0; i < N; ++i) cin >> A[i];
    long long ans = N * (N+1) / 2, cnt = 0;
    mp[cnt]++;
    for (long long i = 0; i < N; ++i) {
        if (A[i] == 1) cnt--;
        else cnt++;
        ans -= mp[cnt]++;
    }
    cout << ans;

    return 0;
}

• Subtask 1: 0 / 10
• Subtask 2: 0 / 15
• Subtask 3: 15 / 15
• Subtask 4: 0 / 70