Submission #526700

#	Time	Username	Problem	Language	Result	Execution time	Memory
526700		joelau	Izbori (COCI22_izbori)	C++14	40 / 110	59 ms	14356 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];
    if (N <= 2000) {
        long long ans = 0;
        for (long long i = 0; i < N; ++i) {
            mp.clear();
            long long most = 0;
            for (long long j = i; j < N; ++j) {
                most = max(most,++mp[A[j]]);
                if (most * 2 > j-i+1) ans++;
            }
        }
        cout << ans;
    }
    else {
        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: 10 / 10
• Subtask 2: 15 / 15
• Subtask 3: 15 / 15
• Subtask 4: 0 / 70