Submission #368289

#	Time	Username	Problem	Language	Result	Execution time	Memory
368289		Vimmer	Tenis (COCI20_tenis)	C++14	110 / 110	82 ms	5096 KiB

tenis

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

//#pragma GCC optimize("-O3")
//#pragma GCC optimize("unroll-loops")
//#pragma GCC optimize("Ofast")

#define N 100501
#define NN 1000500
#define PB push_back
#define M ll(1e9 + 7)
#define all(x) x.begin(), x.end()
#define sz(x) ll(x.size())
#define _ << " " <<
#define pri(x) cout << x << endl
#define endl '\n'
#define F first
#define S second

//using namespace __gnu_pbds;
using namespace std;

typedef long long ll;
typedef unsigned long long ull;
typedef short int si;
typedef long double ld;

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

ll pos[3][N], ans[N], kol[3], skok[4][3], X, Y;

bool cmpr(ll l, ll r)
{
    ll L, R;

    if (pos[l][X] > pos[l][Y])
        L = pos[l][X];
            else L = pos[l][Y];

    if (pos[r][X] > pos[r][Y])
        R = pos[r][X];
            else R = pos[r][Y];

    if (L == R) return l < r;

    return L < R;
}

ll get(ll x)
{
    return min(pos[0][x], min(pos[1][x], pos[2][x]));
}
bool cmp(ll l, ll r)
{
    return get(l) <= get(r);
}

ll opr(ll x)
{
    if (pos[0][x] <= pos[1][x] && pos[0][x] <= pos[2][x]) return 0;

    if (pos[1][x] <= pos[2][x] && pos[1][x] <= pos[0][x]) return 1;

    return 2;
}

ll sk(ll x, ll mn)
{
    ll kl = 0;

    for (ll t = 0; t < 3; t++)
        if (pos[t][x] == mn) kl++;

    return kl;
}

void calc(ll i, ll j)
{
    X = i;
    Y = j;

    ll x = 1e9;

    for (ll t = 0; t < 3; t++)
    {
        if (pos[t][i] > pos[t][j])
            x = min(x, pos[t][j]);
                else x = min(x, pos[t][i]);
    }

    vector <ll> who; who.clear();

    for (ll t = 0; t < 3; t++)
    {
        if (pos[t][i] == x && pos[t][i] < pos[t][j])
        {
            who.PB(t);
        }

        if (pos[t][j] == x && pos[t][j] < pos[t][i])
        {
            who.PB(t);
        }
    }

    sort(all(who), cmpr);

    ll t = who[0];

    kol[t]++;

    if (pos[t][i] > pos[t][j])
        ans[j]++;
            else ans[i]++;
}

int main()
{
    ios_base::sync_with_stdio(0); istream::sync_with_stdio(0); cin.tie(0); cout.tie(0);


//    freopen("pixels.in", "r", stdin);
//    freopen("pixels.out", "w", stdout);

    ll n;

    cin >> n;

    for (ll i = 0; i < 3; i++)
        for (ll j = 0; j < n; j++)
        {
            ll x;

            cin >> x;

            x--;

            pos[i][x] = j;
        }

    vector <ll> vr; vr.clear();

    for (ll i = 0; i < n; i++)
        vr.PB(i);

    sort(all(vr), cmp);

    for (ll i = 0; i < sz(vr); i++)
    {
        ll u = i;

        while (u < sz(vr) && get(vr[i]) == get(vr[u]))
            u++;

        ll l = i, r = u;

        if (l + 1 == r)
        {
            ans[vr[l]] += n - r;

            if (sk(vr[l], pos[opr(vr[l])][vr[l]]) == 1)
                kol[opr(vr[l])] += n - r;

        }
        else if (l + 2 == r)
        {
            ans[vr[l]] = ans[vr[l + 1]] = n - r;

            calc(vr[l], vr[l + 1]);

            ll lt = opr(vr[l]), rt = opr(vr[l + 1]);

            if (sk(vr[l], pos[lt][vr[l]]) == 1)
            {
                kol[lt] += n - r;
            }

            if (sk(vr[l + 1], pos[rt][vr[l + 1]]) == 1)
            {
                kol[rt] += n - r;
            }
        }
        else
        {
            ans[vr[l]] = ans[vr[l + 2]] = ans[vr[l + 1]] = n - r;

            kol[opr(vr[l])] += n - r;
            kol[opr(vr[l + 1])] += n - r;
            kol[opr(vr[l + 2])] += n - r;

            calc(vr[l], vr[l + 1]);

            calc(vr[l], vr[l + 2]);

            calc(vr[l + 1], vr[l + 2]);
        }

        i = u - 1;
    }

    for (ll i = n - 1; i >= 0; i--)
    {
        ll u = i;

        while (u >= 0 && get(vr[u]) == get(vr[i]))
        {
            ll psr = vr[u];

            ll xr = opr(psr);

            if (sk(psr, pos[xr][psr]) == 3)
            {
                for (ll t = 0; t < 3; t++)
                    kol[t] += skok[3][t];
            }
            else if (sk(psr, pos[xr][psr]) == 2)
            {
                ll ps = psr;

                ll val = 0;

                ll mn = pos[xr][psr];

                if (pos[0][ps] == mn && pos[2][ps] == mn)
                    val = 1;
                        else if (pos[1][ps] == mn && pos[2][ps] == mn)
                            val = 2;

                for (ll t = 0; t < 3; t++)
                    kol[t] += skok[val][t];
            }

            u--;
        }

        for (ll t = i; t > u; t--)
        {
            ll ps = vr[t];

            ll x = opr(ps);

            skok[3][x]++;

            if (pos[0][ps] <= pos[1][ps])
                skok[0][0]++;
                    else skok[0][1]++;

            if (pos[0][ps] <= pos[2][ps])
                skok[1][0]++;
                    else skok[1][2]++;

            if (pos[1][ps] <= pos[2][ps])
                skok[2][1]++;
                    else skok[2][2]++;
        }
        i = u + 1;
    }

    ll sum = kol[0] + kol[1] + kol[2];

    if (sum != 1ll * n * (n - 1) / 2)
        assert(0);

    for (ll i = 0; i < 3; i++) cout << kol[i] << " ";

    cout << endl;

    for (ll i = 0; i < n; i++) cout << ans[i] << " ";
}

• Subtask 1: 35 / 35
• Subtask 2: 15 / 15
• Subtask 3: 60 / 60