Submission #302703

#	Time	Username	Problem	Language	Result	Execution time	Memory
302703		Batyr	Carnival Tickets (IOI20_tickets)	C++17	41 / 100	893 ms	71712 KiB

tickets

#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,pos,take;
ll ans,sum,jog;
vector<pii> v;
vi b[maxn][3],d;
vector<vector<int>>s;
priority_queue<pii> g[3];
vector<vi> x;
pii c;

#ifdef LOCAL
void allocate_tickets(vector<vi> a){
	for(int i = 0;i < sz(a);i++){
		for(int j = 0;j < sz(a[0]);j++) cout<<a[i][j]<<' ';
		cout<<'\n';
	}
	return;
}
#endif

void f(int tp,int round){
	memset(us,0,sizeof(us));
	t = 0;
	d.clear();
	while(!g[tp].empty()){
		if(t == n/2) break;
		int x=g[tp].top().ss;
		g[tp].pop();
		us[x] = 1;
		num[x][tp]--;
		b[x][tp].pb(round);
		if(num[x][tp] > 0) d.pb(x);
		t++;
	}
	ans += t;
	for(auto i : d) g[tp].push({num[i][tp],i});
	if(tp==1) an=0;
	else an=1;
	for(int i = 0;i < n;i++){
		if(us[i] or num[i][an]==0) continue;
		num[i][an]--;
		b[i][an].pb(round);
	}
}

ll ch(int pos,int ind){
	v.clear();
	sum = 0,take = n/2-1;
	for(int i = 0;i < n;i++){
		if(i == ind) continue;
		if(x[i][m-1] < pos) sum += pos-x[i][0],take--;
		else if(x[i][0] > pos) sum += x[i][m-1]-pos;
		else v.pb({(pos-x[i][0])-(x[i][m-1]-pos),i});
	}
	
	if(sz(v) < take or take < 0) return -mod;
	
	sort(v.begin(),v.end());
	reverse(v.begin(),v.end());
	for(int i = 0;i < take;i++) sum += pos - x[v[i].ss][0];
	for(int i = take;i < sz(v);i++) sum += x[v[i].ss][m-1] - pos;

	return sum;
}

void put(int pos,int ind){
	v.clear();
	take = n/2-1;
	for(int i = 0;i < n;i++){
		if(i == ind) continue;
		if(x[i][m-1] < pos) s[i][0]=0,take--;
		else if(x[i][0] > pos) s[i][m-1]=0;
		else v.pb({(pos-x[i][0])-(x[i][m-1]-pos),i});
	}	
	
	sort(v.begin(),v.end());
	reverse(v.begin(),v.end());
	for(int i = 0;i < take;i++) s[v[i].ss][0]=0;
	for(int i = take;i < sz(v);i++) s[v[i].ss][m-1]=0;
	
	return;
}

ll find_maximum(int k,vector<vector<int>>_x){
	x = _x;
	n=sz(x);
	m=sz(x[0]);
	
	s.resize(n);
	for(int i = 0;i < n;i++) s[i].resize(m);
	
	if(k==1){
		for(int i = 0;i < n;i++)
			for(int j = 0;j < m;j++) s[i][j] = -1;
		
		for(int i = 0;i < n;i++){
			jog = ch(x[i][0],i);
			if(ans < jog){
				ans = jog;
				c = {i,0};
			}
			jog = ch(x[i][m-1],i);
			if(ans < jog){
				ans = jog;
				c = {i,m-1};
			}
		}
		put(x[c.ff][c.ss],c.ff);
		s[c.ff][c.ss]=0;
		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].push({num[i][0],i});
		if(num[i][1]) g[1].push({num[i][1],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

• Subtask 1: 11 / 11
• Subtask 2: 16 / 16
• Subtask 3: 14 / 14
• Subtask 4: 0 / 14
• Subtask 5: 0 / 12
• Subtask 6: 0 / 23
• Subtask 7: 0 / 10