Submission #1051571

#	Time	Username	Problem	Language	Result	Execution time	Memory
1051571		LittleOrange	Carnival Tickets (IOI20_tickets)	C++17	41 / 100	365 ms	122056 KiB

tickets

#include "tickets.h"
#include <vector>
#include<bits/stdc++.h>
using namespace std;
using ll = long long;
const ll big = 1e18;
struct obj{
	ll v,i,j;
	bool operator<(const obj &o) const{
		return v<o.v;
	}
	bool operator>(const obj &o) const{
		return v>o.v;
	}
};
struct pr{
	ll i,c;
	bool operator<(const pr &o) const{
		return c<o.c;
	}
};
long long find_maximum(int k, std::vector<std::vector<int>> x) {
	ll n = x.size();
	ll m = x[0].size();
	vector<vector<int>> ans(n,vector<int>(m,-1));
	/*if (m == 1){
		vector<ll> v;
		for(ll i = 0;i<n;i++) {
			v.push_back(x[i][0]);
			ans[i][0] = 0;
		}
		allocate_tickets(ans);
		sort(v.begin(),v.end());
		ll sm = 0;
		for(ll i : v) sm += abs(i-v[v.size()/2]);
		return sm;
	}*/
	if (k==1){
		vector<deque<pair<ll,ll>>> a(n);
		for(ll i = 0;i<n;i++){
			for(ll j = 0;j<m;j++){
				a[i].push_back({x[i][j],j});
			}
			sort(a[i].begin(),a[i].end());
		}
		ll sm = 0;
		for(ll r = 0;r<k;r++){
			/*for(ll i = 0;i<n;i++){
				cerr << i << ":";
				for(auto [v,j] : a[i]){
					cerr << " " << v << "," << j;
				}
				cerr << "\n";
			}*/
			vector<pair<ll,ll>> ord;
			for(ll i = 0;i<n;i++){
				ord.push_back({a[i].front().first+a[i].back().first,i});
			}
			sort(ord.begin(),ord.end());
			for(ll i = 0;i<n/2;i++){
				auto [v,j] = a[ord[i].second].front();a[ord[i].second].pop_front();
				ans[ord[i].second][j] = r;
				//cerr << r << " -> " << ord[i].second << " " << j << "\n";
				sm -= v;
			}
			for(ll i = n/2;i<n;i++){
				auto [v,j] = a[ord[i].second].back();a[ord[i].second].pop_back();
				ans[ord[i].second][j] = r;
				//cerr << r << " -> " << ord[i].second << " " << j << "\n";
				sm += v;
			}
		}
		allocate_tickets(ans);
		return sm;
	}
	ll sub3 = 1;
	for(auto &o : x) for(ll i : o) if(i>1) sub3 = 0;
	if(sub3){
		ll sm = 0;
		vector<vector<obj>> v0(n),v1(n);
		for(ll i = 0;i<n;i++){
			for(ll j = 0;j<m;j++){
				if(x[i][j])v1[i].push_back({x[i][j],i,j});
				else v0[i].push_back({x[i][j],i,j});
			}
		}
		priority_queue<pr> q0,q1;
		for(ll i = 0;i<n;i++){
			if (v0[i].size()) q0.push({i,v0[i].size()});
			if (v1[i].size()) q1.push({i,v1[i].size()});
		}
		for(ll j = 0;j<k;j++){
			vector<pr> wait0,wait1;
			vector<ll> u(n,0);
			ll req = n/2;
			while(req&&q1.size()){
				sm++;
				pr p = q1.top();q1.pop();
				p.c = v1[p.i].size();
				if (!p.c) continue;
				obj o = v1[p.i].back();v1[p.i].pop_back();
				p.c--;
				ans[o.i][o.j] = j;
				u[o.i] = 1;
				if(p.c) wait1.push_back(p);
				req--;
			}
			for(ll i = 0;i<n;i++) if(!u[i]){
				if (v0[i].size()){
					obj o = v0[i].back();v0[i].pop_back();
					ans[o.i][o.j] = j;
				}else{
					sm--;
					obj o = v1[i].back();v1[i].pop_back();
					ans[o.i][o.j] = j;
				}
			}
			for(pr p : wait0) q0.push(p);
			for(pr p : wait1) q1.push(p);
		}
		allocate_tickets(ans);
		return sm;
	}
	vector<obj> v1,v2; // big small
	for(ll i = 0;i<n;i++){
		for(ll j = 0;j<m;j++){
			v1.push_back({x[i][j],i,j});
		}
	}
	sort(v1.begin(),v1.end());
	v2 = v1;
	reverse(v2.begin(),v2.end());
	vector<vector<obj>> g1(n),g2(n);
	ll cnt = n/2*k;
	ll sm = 0;
	for(ll i = 0;i<cnt;i++){
		obj o = v1.back();v1.pop_back();
		sm += o.v;
		g1[o.i].push_back(o);
	}
	for(ll i = 0;i<cnt;i++){
		obj o = v2.back();v2.pop_back();
		sm -= o.v;
		g2[o.i].push_back(o);
	}
	ll bad = 0;
	for(ll i = 0;i<n;i++) if(g1[i].size()+g2[i].size()>k) bad++;
	while(bad){
		obj o1 = {big,-1,-1}, o2 = {-big,-1,-1};
		for(ll i = 0;i<n;i++) if(g1[i].size()+g2[i].size()>k){
			o1 = min(o1,g1[i].back());
			o2 = max(o2,g2[i].back());
		}
		ll d1 = o1.v-v1.back().v;
		ll d2 = v2.back().v-o2.v;
		if (d1<=d2){
			sm -= d1;
			g1[o1.i].pop_back();
			g1[v1.back().i].push_back(v1.back());
			v1.pop_back();
		}else{
			sm -= d2;
			g2[o2.i].pop_back();
			g2[v2.back().i].push_back(v2.back());
			v2.pop_back();
		}
		bad = 0;
		for(ll i = 0;i<n;i++) if(g1[i].size()+g2[i].size()>k) bad++;
	}
	vector<pair<ll,ll>> ord;
	for(ll i = 0;i<n;i++) ord.push_back({g1[i].size(),i});
	sort(ord.begin(),ord.end(),greater<pair<ll,ll>>());
	for(ll j = 0;j<k;j++){
		ll c1 = n/2;
		for(auto [_,i] : ord){
			obj x;
			if (g1[i].size()&&c1--){
				x = g1[i].back();g1[i].pop_back();
			}else{
				x = g2[i].back();g2[i].pop_back();
			}
			ans[i][x.j] = j;
		}
	}
	allocate_tickets(ans);
	return sm;
}

Compilation message (stderr)

tickets.cpp: In function 'long long int find_maximum(int, std::vector<std::vector<int> >)':
tickets.cpp:89:43: warning: narrowing conversion of '(& v0.std::vector<std::vector<obj> >::operator[](((std::vector<std::vector<obj> >::size_type)i)))->std::vector<obj>::size()' from 'std::vector<obj>::size_type' {aka 'long unsigned int'} to 'll' {aka 'long long int'} [-Wnarrowing]
   89 |    if (v0[i].size()) q0.push({i,v0[i].size()});
      |                                 ~~~~~~~~~~^~
tickets.cpp:90:43: warning: narrowing conversion of '(& v1.std::vector<std::vector<obj> >::operator[](((std::vector<std::vector<obj> >::size_type)i)))->std::vector<obj>::size()' from 'std::vector<obj>::size_type' {aka 'long unsigned int'} to 'll' {aka 'long long int'} [-Wnarrowing]
   90 |    if (v1[i].size()) q1.push({i,v1[i].size()});
      |                                 ~~~~~~~~~~^~
tickets.cpp:147:52: warning: comparison of integer expressions of different signedness: 'std::vector<obj>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
  147 |  for(ll i = 0;i<n;i++) if(g1[i].size()+g2[i].size()>k) bad++;
      |                           ~~~~~~~~~~~~~~~~~~~~~~~~~^~
tickets.cpp:150:53: warning: comparison of integer expressions of different signedness: 'std::vector<obj>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
  150 |   for(ll i = 0;i<n;i++) if(g1[i].size()+g2[i].size()>k){
      |                            ~~~~~~~~~~~~~~~~~~~~~~~~~^~
tickets.cpp:168:53: warning: comparison of integer expressions of different signedness: 'std::vector<obj>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
  168 |   for(ll i = 0;i<n;i++) if(g1[i].size()+g2[i].size()>k) bad++;
      |                            ~~~~~~~~~~~~~~~~~~~~~~~~~^~

• 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