Submission #1166083

#	Time	Username	Problem	Language	Result	Execution time	Memory
1166083		King87arshia	Izbori (COCI22_izbori)	C++17	25 / 110	3092 ms	840 KiB

Main

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>

using namespace std;
using namespace __gnu_pbds;

typedef tree<pair<int, int>, null_type, less<pair<int, int>>, rb_tree_tag, tree_order_statistics_node_update> ordered_set;

#pragma GCC optimize ("Ofast,unroll-loops")
#pragma GCC target ("avx2")

const int N = 2e5 + 5, T = 2e5;

vector<int> big;
int ans = 0, n, a[N];
map<int, int> mp, mp2;
map<int, bool> mark;

int main() {
    scanf("%lld", &n);
    for (int i = 0; i < n; i++) {
        scanf("%lld", &a[i]);
        mp[a[i]]++;
    }
    for (auto &[x, y] : mp) {
        if (y >= T) big.push_back(x);
        else mark[x] = true;
    }
    for (long long x : big) {
        ordered_set s;
        long long fake[n + 5];
        int sum = 0;
        for (int i = 0; i < n; i++)
            fake[i] = (a[i] == x ? 1 : -1);
        s.insert({0, -1});
        for (int i = 0; i < n; i++) {
            sum += fake[i];
            ans += s.order_of_key({sum, INT_MIN});
            s.insert({sum, i});
        }
    }
    for (int i = 0; i < n; i++) {
        pair<int, int> mx = {0, 0};
        mp2.clear();
        for (int sz = 1; sz <= 2 * T; sz++) {
            if (i + sz - 1 >= n)
                break;
            mp2[a[i + sz - 1]]++;
            long long x = mp2[a[i + sz - 1]];
            if (mx.first < x)
                mx = {x, a[i + sz - 1]};
            if (mx.first * 2 > sz && mp[mx.second] < T)
                ans++;
        }
    }
    printf("%lld\n", ans);
    return 0;
}

Compilation message (stderr)

Main.cpp: In function 'int main()':
Main.cpp:21:15: warning: format '%lld' expects argument of type 'long long int*', but argument 2 has type 'int*' [-Wformat=]
   21 |     scanf("%lld", &n);
      |            ~~~^   ~~
      |               |   |
      |               |   int*
      |               long long int*
      |            %d
Main.cpp:23:19: warning: format '%lld' expects argument of type 'long long int*', but argument 2 has type 'int*' [-Wformat=]
   23 |         scanf("%lld", &a[i]);
      |                ~~~^   ~~~~~
      |                   |   |
      |                   |   int*
      |                   long long int*
      |                %d
Main.cpp:57:16: warning: format '%lld' expects argument of type 'long long int', but argument 2 has type 'int' [-Wformat=]
   57 |     printf("%lld\n", ans);
      |             ~~~^     ~~~
      |                |     |
      |                |     int
      |                long long int
      |             %d
Main.cpp:21:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   21 |     scanf("%lld", &n);
      |     ~~~~~^~~~~~~~~~~~
Main.cpp:23:14: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   23 |         scanf("%lld", &a[i]);
      |         ~~~~~^~~~~~~~~~~~~~~

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