oj.uz

Submission #366452

# Submission time Handle Problem Language Result Execution time Memory
366452 2021-02-14T08:25:27 Z VEGAnn Tenis (COCI20_tenis) C++14
110 / 110
 122 ms 7808 KB 
#include <bits/stdc++.h>
#define i2 array<int,2>
using namespace std;
typedef long long ll;
const int N = 100100;
const int oo = 2e9;
int ok1[3][8], ok2[3][8][8];
ll win[N], crt[3];
int a[3][N], n, loc[3][N];
int MIN[8][N], nm[N];

void calc(int x, int y){
    i2 mn = {oo, oo};

    for (int j = 0; j < 3; j++)
        mn = min(mn, {min(loc[j][x], loc[j][y]), max(loc[j][x], loc[j][y])});

    for (int j = 0; j < 3; j++) {
        i2 cr = {min(loc[j][x], loc[j][y]), max(loc[j][x], loc[j][y])};

        if (mn == cr){
            if (loc[j][x] > loc[j][y])
                swap(x, y);

            win[x]++;
            crt[j]++;

            return;
        }
    }

}

bool cmp(int _x, int _y){
    return MIN[7][_x] > MIN[7][_y];
}

void upd_ans(int x){
    int sub = 0;

    for (int i = 0; i < 3; i++)
        if (loc[i][x] == MIN[7][x])
            sub |= (1 << i);

    {
        // 0
        if (MIN[7][x] == loc[0][x]) {
            crt[0] += ok1[0][sub];
            win[x] += ok1[0][sub];
        }
    }

    {
        // 1
        if (MIN[7][x] == loc[1][x]) {
            crt[1] += ok2[1][sub][(1 & sub)];
            win[x] += ok2[1][sub][(1 & sub)];
        }
    }

    {
        // 2
        if (MIN[7][x] == loc[2][x]) {
            crt[2] += ok2[2][sub][(3 & sub)];
            win[x] += ok2[2][sub][(3 & sub)];
        }
    }
}

void insert(int x){
    for (int i = 0; i < 3; i++)
    for (int sub = 0; sub < 8; sub++){
        ok1[i][sub] += bool(MIN[sub][x] == loc[i][x]);

        for (int sb = 0; sb < 8; sb++)
            if ((sb & sub) == sb)
                ok2[i][sub][sb] += bool(MIN[sub][x] == loc[i][x] && MIN[sub][x] < MIN[sb][x]);
    }
}

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

#ifdef _LOCAL
    freopen("in.txt","r",stdin);
#endif // _LOCAL

    cin >> n;

    for (int j = 0; j < 3; j++)
    for (int i = 1; i <= n; i++) {
        cin >> a[j][i];

        loc[j][a[j][i]] = i;
    }

    for (int i = 1; i <= n; i++)
    for (int msk = 0; msk < 8; msk++){
        int res = oo;

        for (int j = 0; j < 3; j++)
            if (msk & (1 << j))
                res = min(res, loc[j][i]);

        MIN[msk][i] = res;

        nm[i] = i;
    }

    sort(nm + 1, nm + n + 1, cmp);

    for (int i = 1; i <= n; ){
        int j = i;

        while (j <= n && MIN[7][nm[j]] == MIN[7][nm[i]]) {
            upd_ans(nm[j]);
            j++;
        }

        for (int i1 = i; i1 < j; i1++)
        for (int I2 = i1 + 1; I2 < j; I2++)
            calc(nm[i1], nm[I2]);

        for (int i1 = i; i1 < j; i1++)
            insert(nm[i1]);

        i = j;
    }

    for (int i = 0; i < 3; i++)
        cout << crt[i] << " ";
    cout << '\n';

    for (int i = 1; i <= n; i++)
        cout << win[i] << " ";

    return 0;
}

Subtask #1
35.0 / 35.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 364 KB Output is correct
2 Correct 1 ms 364 KB Output is correct
3 Correct 1 ms 492 KB Output is correct

Subtask #2
15.0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 364 KB Output is correct
2 Correct 1 ms 364 KB Output is correct
3 Correct 1 ms 492 KB Output is correct
4 Correct 4 ms 640 KB Output is correct

Subtask #3
60.0 / 60.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 364 KB Output is correct
2 Correct 1 ms 364 KB Output is correct
3 Correct 1 ms 492 KB Output is correct
4 Correct 4 ms 640 KB Output is correct
5 Correct 46 ms 3260 KB Output is correct
6 Correct 72 ms 4844 KB Output is correct
7 Correct 104 ms 6380 KB Output is correct
8 Correct 122 ms 7788 KB Output is correct
9 Correct 119 ms 7788 KB Output is correct
10 Correct 121 ms 7808 KB Output is correct