//Original Code:
//#include <self/utility>
//#include <self/debug>
//using namespace std;
//int n;
//int p[3][100005];
//int q[3][100005];
//int ap[100005];
//int res[100005];
//ll ans[3];
//int vis[100005][8];
//int f[8];
//vector<int> tm[100005];
//
//inline int match(int x,int y)//????
//{
// vector<tuple<int,int,int> > v;
// for(int i=0;i<3;i++)
// {
// v.emplace_back(min(q[i][x],q[i][y]),max(q[i][x],q[i][y]),i);
// }
// sort(all(v));
// return get<2>(v[0]);
//}
//
//inline int match(int x,int y,int z)//????
//{
// if(q[z][x]<q[z][y]) return x;else return y;
//}
//
//int main()
//{
// // freopen("input.txt","r",stdin);
// cin>>n;
// for(int i=0;i<3;i++)
// {
// for(int j=0;j<n;j++)
// {
// cin>>p[i][j];
// p[i][j]--;
// q[i][p[i][j]]=j;
// }
// }
// for(int j=0;j<n;j++)
// {
// vector<int> t;
// for(int i=0;i<3;i++)
// {
// t.push_back(p[i][j]);
// }
// sort(all(t));
// t.erase(unique(all(t)),t.end());
// for(auto x:t)
// {
// int m=0;
// for(int i=0;i<3;i++)
// {
// if(p[i][j]==x) m|=(1<<i);
// }
// tm[x].push_back(m);
// }
// }
// int naps=n;
// for(int j=0;j<n;j++)
// {
// vector<int> t;
// for(int i=0;i<3;i++)
// {
// if(!ap[p[i][j]])
// {
// ap[p[i][j]]=1;
// naps--;
// t.push_back(p[i][j]);
// }
// }
// for(auto x:t)
// {
// for(int mask=1;mask<8;mask++)
// {
// for(auto m:tm[x])
// {
// int t=mask & m;
// if(!t) continue;
// int id=__builtin_ctz(t);
// ans[id]+=f[mask];
// break;
// }
// }
// }
// for(auto x:t)
// {
// int m=0;
// for(int i=0;i<3;i++)
// {
// if(p[i][j]==x) m|=(1<<i);
// }
// f[m]++;
// }
// for(int a=0;a<(int)t.size();a++)
// {
// for(int b=a+1;b<(int)t.size();b++)
// {
// int w=match(t[a],t[b]);
// // debug(w);
// ans[w]++;
// res[match(t[a],t[b],w)]++;
// }
// }
// for(auto x:t)
// {
// res[x]+=naps;
// }
// }
// for(int i=0;i<3;i++) cout<<ans[i]<<' ';
// cout<<endl;
// for(int i=0;i<n;i++)
// {
// cout<<res[i]<<' ';
// }
// cout<<endl;
// return 0;
//}
//substituted with C++ Inliner
#ifndef _SELF_UTILITY
#define _SELF_UTILITY 1
#include <numeric>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <stdio.h>
#include <stdlib.h>
#include <vector>
#include <map>
#include <queue>
#include <set>
#include <string>
#include <string.h>
#include <stack>
#include <assert.h>
#include <bitset>
#include <time.h>
#define Endl endl
#define mp make_pair
#define mt make_tuple
#define ll long long
#define ull unsigned long long
#define pii pair<int,int>
#define over(A) {cout<<A<<endl;exit(0);}
#define all(A) A.begin(),A.end()
#define quickcin ios_base::sync_with_stdio(false);
#ifdef __TAKE_MOD
int mod;
#else
#ifdef __TAKE_CONST_MOD
const int mod=__TAKE_CONST_MOD;
#else
const int mod=1000000007;
#endif
#endif
const int gmod=3;
const int inf=1039074182;
#ifdef __TAKE_CONST_EPS
const double eps=__TAKE_CONST_EPS;
#else
const double eps=1e-9;
#endif
const double pi=3.141592653589793238462643383279;
const ll llinf=2LL*inf*inf;
template <typename T1,typename T2> inline void chmin(T1 &x,T2 b) {if(b<x) x=b;}
template <typename T1,typename T2> inline void chmax(T1 &x,T2 b) {if(b>x) x=b;}
inline void chadd(int &x,int b) {x+=b-mod;x+=(x>>31 & mod);}
template <typename T1,typename T2> inline void chadd(T1 &x,T2 b) {x+=b;if(x>=mod) x-=mod;}
template <typename T1,typename T2> inline void chmul(T1 &x,T2 b) {x=1LL*x*b%mod;}
template <typename T1,typename T2> inline void chmod(T1 &x,T2 b) {x%=b,x+=b;if(x>=b) x-=b;}
template <typename T> inline T mabs(T x) {return (x<0?-x:x);}
using namespace std;
#endif
#ifndef _SELF_DEBUG
#define _SELF_DEBUG 1
#ifndef _SELF_OPERATOR
#define _SELF_OPERATOR 1
using namespace std;
template <typename T>
ostream & operator<<(ostream &cout, const vector<T> &vec)
{
cout << "{";
for (int i = 0; i < (int)vec.size(); i++)
{
cout << vec[i];
if (i != (int)vec.size() - 1)
cout << ',';
}
cout << "}";
return cout;
}
template <typename T1, typename T2>
ostream &operator<<(ostream &cout, pair<T1, T2> p)
{
cout << "(" << p.first << ',' << p.second << ")";
return cout;
}
template <typename T, typename T2>
ostream &operator<<(ostream &cout, set<T, T2> s)
{
vector<T> t;
for (auto x : s)
t.push_back(x);
cout << t;
return cout;
}
template <typename T, typename T2>
ostream &operator<<(ostream &cout, multiset<T, T2> s)
{
vector<T> t;
for (auto x : s)
t.push_back(x);
cout << t;
return cout;
}
template <typename T>
ostream &operator<<(ostream &cout, queue<T> q)
{
vector<T> t;
while (q.size())
{
t.push_back(q.front());
q.pop();
}
cout << t;
return cout;
}
template <typename T1, typename T2, typename T3>
ostream &operator<<(ostream &cout, map<T1, T2, T3> m)
{
for (auto &x : m)
{
cout << "Key: " << x.first << ' ' << "Value: " << x.second << endl;
}
return cout;
}
template <typename T1, typename T2>
void operator+=(pair<T1, T2> &x,const pair<T1, T2> y)
{
x.first += y.first;
x.second += y.second;
}
template <typename T1,typename T2>
pair<T1,T2> operator + (const pair<T1,T2> &x,const pair<T1,T2> &y)
{
return make_pair(x.first+y.first,x.second+y.second);
}
template <typename T1,typename T2>
pair<T1,T2> operator - (const pair<T1,T2> &x,const pair<T1,T2> &y)
{
return mp(x.first-y.first,x.second-y.second);
}
template <typename T1, typename T2>
pair<T1, T2> operator-(pair<T1, T2> x)
{
return make_pair(-x.first, -x.second);
}
template <typename T>
vector<vector<T>> operator~(vector<vector<T>> vec)
{
vector<vector<T>> v;
int n = vec.size(), m = vec[0].size();
v.resize(m);
for (int i = 0; i < m; i++)
{
v[i].resize(n);
}
for (int i = 0; i < m; i++)
{
for (int j = 0; j < n; j++)
{
v[i][j] = vec[j][i];
}
}
return v;
}
#endif
#include <sstream>
void print0x(int x)
{
std::vector <int> vec;
while(x)
{
vec.push_back(x&1);
x>>=1;
}
std::reverse(vec.begin(),vec.end());
for(int i=0;i<(int)vec.size();i++)
{
std::cout<<vec[i];
}
std::cout<<' ';
}
template <typename T>
void print0x(T x,int len)
{
std::vector <int> vec;
while(x)
{
std::cout<<(x&1);
x>>=1;
len--;
// vec.push_back(x&1);
// x>>=1;
}
while(len--) cout<<0;
// reverse(vec.begin(),vec.end());
// for(int i=(int)vec.size();i<len;i++)
// {
// putchar('0');
// }
// for(size_t i=0;i<vec.size();i++)
// {
// std::cout<<vec[i];
// }
// std::cout<<' ';
}
vector<string> vec_splitter(string s) {
s += ',';
vector<string> res;
while(!s.empty()) {
res.push_back(s.substr(0, s.find(',')));
s = s.substr(s.find(',') + 1);
}
return res;
}
void debug_out(
vector<string> __attribute__ ((unused)) args,
__attribute__ ((unused)) int idx,
__attribute__ ((unused)) int LINE_NUM) { cerr << endl; }
template <typename Head, typename... Tail>
void debug_out(vector<string> args, int idx, int LINE_NUM, Head H, Tail... T) {
if(idx > 0) cerr << ", "; else cerr << "Line(" << LINE_NUM << ") ";
stringstream ss; ss << H;
cerr << args[idx] << " = " << ss.str();
debug_out(args, idx + 1, LINE_NUM, T...);
}
#define debug(...) debug_out(vec_splitter(#__VA_ARGS__), 0, __LINE__, __VA_ARGS__)
#endif
using namespace std;
int n;
int p[3][100005];
int q[3][100005];
int ap[100005];
int res[100005];
ll ans[3];
int vis[100005][8];
int f[8];
vector<int> tm[100005];
inline int match(int x,int y)//????
{
vector<tuple<int,int,int> > v;
for(int i=0;i<3;i++)
{
v.emplace_back(min(q[i][x],q[i][y]),max(q[i][x],q[i][y]),i);
}
sort(all(v));
return get<2>(v[0]);
}
inline int match(int x,int y,int z)//????
{
if(q[z][x]<q[z][y]) return x;else return y;
}
int main()
{
// // freopen("input.txt","r",stdin);
cin>>n;
for(int i=0;i<3;i++)
{
for(int j=0;j<n;j++)
{
cin>>p[i][j];
p[i][j]--;
q[i][p[i][j]]=j;
}
}
for(int j=0;j<n;j++)
{
vector<int> t;
for(int i=0;i<3;i++)
{
t.push_back(p[i][j]);
}
sort(all(t));
t.erase(unique(all(t)),t.end());
for(auto x:t)
{
int m=0;
for(int i=0;i<3;i++)
{
if(p[i][j]==x) m|=(1<<i);
}
tm[x].push_back(m);
}
}
int naps=n;
for(int j=0;j<n;j++)
{
vector<int> t;
for(int i=0;i<3;i++)
{
if(!ap[p[i][j]])
{
ap[p[i][j]]=1;
naps--;
t.push_back(p[i][j]);
}
}
for(auto x:t)
{
for(int mask=1;mask<8;mask++)
{
for(auto m:tm[x])
{
int t=mask & m;
if(!t) continue;
int id=__builtin_ctz(t);
ans[id]+=f[mask];
break;
}
}
}
for(auto x:t)
{
int m=0;
for(int i=0;i<3;i++)
{
if(p[i][j]==x) m|=(1<<i);
}
f[m]++;
}
for(int a=0;a<(int)t.size();a++)
{
for(int b=a+1;b<(int)t.size();b++)
{
int w=match(t[a],t[b]);
// debug(w);
ans[w]++;
res[match(t[a],t[b],w)]++;
}
}
for(auto x:t)
{
res[x]+=naps;
}
}
for(int i=0;i<3;i++) cout<<ans[i]<<' ';
cout<<endl;
for(int i=0;i<n;i++)
{
cout<<res[i]<<' ';
}
cout<<endl;
return 0;
}
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Correct |
2 ms |
2636 KB |
Output is correct |
| 2 |
Correct |
2 ms |
2636 KB |
Output is correct |
| 3 |
Correct |
3 ms |
2636 KB |
Output is correct |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Correct |
2 ms |
2636 KB |
Output is correct |
| 2 |
Correct |
2 ms |
2636 KB |
Output is correct |
| 3 |
Correct |
3 ms |
2636 KB |
Output is correct |
| 4 |
Correct |
6 ms |
2892 KB |
Output is correct |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Correct |
2 ms |
2636 KB |
Output is correct |
| 2 |
Correct |
2 ms |
2636 KB |
Output is correct |
| 3 |
Correct |
3 ms |
2636 KB |
Output is correct |
| 4 |
Correct |
6 ms |
2892 KB |
Output is correct |
| 5 |
Correct |
77 ms |
5380 KB |
Output is correct |
| 6 |
Correct |
105 ms |
6724 KB |
Output is correct |
| 7 |
Correct |
173 ms |
8152 KB |
Output is correct |
| 8 |
Correct |
214 ms |
9488 KB |
Output is correct |
| 9 |
Correct |
188 ms |
9400 KB |
Output is correct |
| 10 |
Correct |
154 ms |
9452 KB |
Output is correct |
