Submission #481530

#	Time	Username	Problem	Language	Result	Execution time	Memory
481530		CSQ31	Tenis (COCI20_tenis)	C++17	55 / 110	94 ms	10548 KiB

tenis

#pragma GCC optimize("Ofast") 
#include<bits/stdc++.h>
using namespace std;
#define pb push_back
#define fi first
#define se second
#define sz(a) (int)(a.size())
#define all(a) a.begin(),a.end()
#define lb lower_bound
#define ub upper_bound
#define owo ios_base::sync_with_stdio(0);cin.tie(0);
#define MOD (ll)(998244353)
#define INF (ll)(1e18)
#define debug(...) fprintf(stderr, __VA_ARGS__),fflush(stderr)
#define time__(d) for(long blockTime = 0; (blockTime == 0 ? (blockTime=clock()) != 0 : false);\
debug("%s time : %.4fs\n", d, (double)(clock() - blockTime) / CLOCKS_PER_SEC))
typedef long long int ll;
typedef long double ld;
typedef pair<ll,ll> PII;
typedef pair<int,int> pii;
typedef vector<vector<int>> vii;
typedef vector<vector<ll>> VII;
ll gcd(ll a,ll b){if(!b)return a;else return gcd(b,a%b);}
const int MAXN = 1e5+1;
int a[3][MAXN],ans[MAXN],w[MAXN],cr[3];
int mask[MAXN];
vii pmask(100),tmask(100),same(MAXN);
//if 1-1 exist,we fine just take first 1,1
//only case we need to worry about is 2 bits + its complement
//this case we need info on the two zeros in the complement
//we need an extra mask for each mask with two zero
int main()
{
	owo
	int n;
	cin>>n;
	for(int j=0;j<3;j++){
	   for(int i=0;i<n;i++){
		   int c;
		   cin>>c;
		   a[j][c-1]=i;
	   }
   }
   for(int i=0;i<n;i++){
	   w[i] = min({a[0][i],a[1][i],a[2][i]});
	   int d0=-1,d1=-1;
	   for(int j=0;j<3;j++){
		   if(w[i] == a[j][i]){
			   mask[i]+=(1<<j);
			   if(d0==-1)d0=j;
			   else d1=j;
		   }
	   }
	   pmask[mask[i]].pb(w[i]);
	   if(__builtin_popcount(mask[i]) == 2){
		   if(d0 < d1)tmask[mask[i]].pb(w[i]);
		   else tmask[mask[i]+8].pb(w[i]);
	   }   
	   same[w[i]].pb(i);
   }
   for(int i=0;i<20;i++)sort(all(pmask[i]));
   
   for(int i=0;i<n;i++){
	   for(int j=0;j<8;j++){
		   ans[i]+=sz(pmask[j]) - (ub(all(pmask[j]),w[i]) - pmask[j].begin());
	   }
   }
   //non strict wins
   for(int i=0;i<n;i++){
	   for(int j=0;j<sz(same[i]);j++){
		   for(int k=j+1;k<sz(same[i]);k++){
			   int v = same[i][j];
			   int u = same[i][k];
			   int wv = a[0][v],wu = a[0][u],id = 0;
			   for(int l=1;l<3;l++){
				   if(min(a[l][v],a[l][u])<min(wv,wu)){
					   id = l;
					   wv = a[l][v];
					   wu = a[l][u];
				   }else if(min(a[l][v],a[l][u]) == min(wv,wu)){
					   if(max(a[l][v],a[l][u]) < max(wv,wu)){
						   id = l;
						   wv = a[l][v];
						   wu = a[l][u];
					   }
				   }
			   }
			   cr[id]++;
			   if(wv < wu)ans[v]++;
			   else ans[u]++;
		   }
	   }
   }
   cout<<cr[0]<<" "<<cr[1]<<" "<<cr[2]<<'\n';
   for(int i=0;i<n;i++)cout<<ans[i]<<" ";
}

• Subtask 1: 17.5 / 35
• Subtask 2: 7.5 / 15
• Subtask 3: 30 / 60