Submission #476376

# Submission time Handle Problem Language Result Execution time Memory
476376 2021-09-26T08:59:31 Z MonkeyKing Tenis (COCI20_tenis) C++17
110 / 110
214 ms 9488 KB
//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;
}
# Verdict Execution time Memory 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
# Verdict Execution time Memory 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
# Verdict Execution time Memory 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