| # | Time | Username | Problem | Language | Result | Execution time | Memory |
|---|---|---|---|---|---|---|---|
| 301528 | Batyr | 카니발 티켓 (IOI20_tickets) | C++17 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include<bits/stdc++.h>
#ifndef LOCAL
#include "tickets.h"
#endif
using namespace std;
#define pb push_back
#define ff first
#define ss second
#define sz(a) (int)a.size()
typedef long long ll;
typedef vector<int> vi;
typedef pair<int,int> pii;
const ll inf=1e18;
const int mod=1e9+7;
const int maxn=2e3+5;
int n,m,num[maxn][3],cnt[3];
int us[maxn],t,an;
ll ans;
vi b[maxn][3],d,v;
vector<vector<int>>s;
set <int>g[3];
#ifdef LOCAL
void allocate_tickets(vector<vi> a){
return;
}
#endif
void f(int tp,int round){
memset(us,0,sizeof(us));
t = 0;
d.clear();
for(auto i : g[tp]){
if(t == n/2) break;
us[i] = 1;
b[i][tp].pb(round);
num[i][tp]--;
if(num[i][tp]==0) d.pb(i);
t++;
}
ans += t;
for(auto i : d) g[tp].erase(i);
d.clear();
if(tp==1) an=0;
else an=1;
for(auto i : g[an]){
if(us[i]) continue;
b[i][an].pb(round);
num[i][an]--;
if(num[i][an]==0) d.pb(i);
taken++;
}
for(auto i : d) g[an].erase(i);
}
ll find_maximum(int k,vector<vector<int>>x){
n=sz(x);
m=sz(x[0]);
s.resize(n);
for(int i = 0;i < n;i++) s[i].resize(m);
if(m==1){
for(int i = 0;i < n;i++) v.pb(x[i][0]);
sort(v.begin(),v.end());
for(int i = 0;i < n;i++) ans += abs(v[n/2]-v[i]);
allocate_tickets(s);
return ans;
}
for(int i = 0;i < n;i++){
for(int j = 0;j < m;j++) num[i][x[i][j]]++;
if(num[i][0]) g[0].insert(i);
if(num[i][1]) g[1].insert(i);
cnt[0] += num[i][0];
cnt[1] += num[i][1];
}
for(int i = 0;i < k;i++){
if(cnt[0] < cnt[1]) f(0,i);
else f(1,i);
}
for(int i = 0;i < n;i++){
for(int j = 0;j < m;j++){
if(sz(b[i][x[i][j]])){
s[i][j] = b[i][x[i][j]].back();
b[i][x[i][j]].pop_back();
}
else s[i][j] = -1;
}
}
allocate_tickets(s);
return ans;
}
#ifdef LOCAL
int main(){
int n,m,k;
cin >> n >> m >> k;
vector<vi> c;
c.resize(n);
for(int i = 0;i < n;i++){
c[i].resize(m);
for(int j = 0;j < m;j++) cin >> c[i][j];
}
cout<<find_maximum(k,c)<<'\n';
}
#endif