#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) int(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;
int pos[3][N], ans[N], kol[3], skok[4][3], X, Y;
bool cmpr(int l, int r)
{
int 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;
}
int get(int x)
{
return min(pos[0][x], min(pos[1][x], pos[2][x]));
}
bool cmp(int l, int r)
{
return get(l) <= get(r);
}
int opr(int 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;
}
int sk(int x, int mn)
{
int kl = 0;
for (int t = 0; t < 3; t++)
if (pos[t][x] == mn) kl++;
return kl;
}
int calc(int i, int j)
{
X = i;
Y = j;
multiset <int> se; se.clear();
for (int t = 0; t < 3; t++)
{
if (pos[t][i] > pos[t][j])
se.insert(pos[t][j]);
else se.insert(pos[t][i]);
}
vector <int> who; who.clear();
int x = *se.begin();
for (int 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);
int 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("1.in", "r", stdin);
//
// freopen("pixels.in", "r", stdin);
// freopen("pixels.out", "w", stdout);
int n;
cin >> n;
for (int i = 0; i < 3; i++)
for (int j = 0; j < n; j++)
{
int x;
cin >> x;
x--;
pos[i][x] = j;
}
vector <int> vr; vr.clear();
for (int i = 0; i < n; i++)
vr.PB(i);
sort(all(vr), cmp);
for (int i = 0; i < sz(vr); i++)
{
int u = i;
while (u < sz(vr) && get(vr[i]) == get(vr[u]))
u++;
int l = i, r = u;
if (l + 1 == r)
{
ans[vr[l]] += n - r;
}
else if (l + 2 == r)
{
ans[vr[l]] = ans[vr[l + 1]] = n - r;
calc(vr[l], vr[l + 1]);
int lt = opr(l), rt = opr(r);
if (sk(l, pos[lt][l]) == 1)
{
kol[lt] += n - r;
}
else
{
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 (int i = n - 1; i >= 0; i--)
{
int u = i;
while (u >= 0 && get(vr[u]) == get(vr[i]))
{
int xr = opr(vr[u]);
int ps = vr[u];
if (sk(vr[u], xr) == 3)
{
for (int t = 0; t < 3; t++)
kol[t] += skok[3][t];
}
else if (sk(vr[u], xr) == 2)
{
int ps = vr[u];
int val = 0;
if (pos[0][ps] == xr && pos[2][ps] == xr)
val = 1;
else if (pos[1][ps] == xr && pos[2][ps] == xr)
val = 2;
for (int t = 0; t < 3; t++)
kol[t] += skok[val][t];
}
u--;
}
for (int t = i; t > u; t--)
{
int ps = vr[t];
int x = opr(vr[t]);
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;
}
for (int i = 0; i < 3; i++) cout << kol[i] << " ";
cout << endl;
for (int i = 0; i < n; i++) cout << ans[i] << " ";
}
Compilation message
tenis.cpp: In function 'int calc(int, int)':
tenis.cpp:118:1: warning: no return statement in function returning non-void [-Wreturn-type]
118 | }
| ^
tenis.cpp: In function 'int main()':
tenis.cpp:208:17: warning: unused variable 'ps' [-Wunused-variable]
208 | int ps = vr[u];
| ^~
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Runtime error |
1 ms |
492 KB |
Execution killed with signal 11 |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Runtime error |
1 ms |
492 KB |
Execution killed with signal 11 |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Runtime error |
1 ms |
492 KB |
Execution killed with signal 11 |
| 2 |
Halted |
0 ms |
0 KB |
- |
