Submission #830159

#	Time	Username	Problem	Language	Result	Execution time	Memory
830159		Baytoro	Carnival Tickets (IOI20_tickets)	C++17	41 / 100	668 ms	84084 KiB

tickets

#include "tickets.h"
//#include "grader.cpp"
#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define fr first
#define sc second
//#define int long long
long long find_maximum(int k, vector<vector<int>> x) {
	int n=x.size(),m=x[0].size();
	if(k==1){
		vector<vector<int>> ans(n,vector<int>(m,-1));
		vector<pair<int,int>> v(n);
		vector<int> a;
		for(int i=0;i<n;i++){
			v[i]={x[i][0]+x[i][m-1],i};
		}
		sort(v.begin(),v.end());
		for(int i=0;i<n/2;i++){
			ans[v[i].sc][0]=0;
			a.pb(x[v[i].sc][0]);
		}
		for(int i=n/2;i<n;i++){
			ans[v[i].sc][m-1]=0;
			a.pb(x[v[i].sc][m-1]);
		}
		allocate_tickets(ans);
		sort(a.begin(),a.end());
		long long s=(a[n/2]),res=0;
		for(int i=0;i<n;i++) res+=abs(s-a[i]);
		return res;
	}
	else if(k==m){
		vector<vector<int>> ans(n,vector<int>(m,-1));
		long long res=0;
		vector<int> l(n),r(n,m-1),id(n),a(n);
		iota(id.begin(),id.end(),0);
		vector<pair<int,int>> v;
		for(int i=0;i<n;i++){
			for(int j=0;j<m;j++)
				v.pb({x[i][j],i});
		}
		sort(v.begin(),v.end());
		for(int i=0;i<(int)v.size()/2;i++)	
			a[v[i].sc]++;
		for(int t=0;t<k;t++){
			sort(id.begin(),id.end(),[&](int i, int j){
				return a[i]>a[j];
			});
			long long X=0,Y=0;
			for(int i=0;i<n/2;i++){
				X+=x[id[i]][l[id[i]]];
				ans[id[i]][l[id[i]]]=t;
				a[id[i]]--;
				l[id[i]]++;
			}
			for(int i=n/2;i<n;i++){
				Y+=x[id[i]][r[id[i]]];
				ans[id[i]][r[id[i]]]=t;
				r[id[i]]--;
			}
			res+=Y-X;
		}
		
		allocate_tickets(ans);
		return res;
	}
}

Compilation message (stderr)

tickets.cpp: In function 'long long int find_maximum(int, std::vector<std::vector<int> >)':
tickets.cpp:68:1: warning: control reaches end of non-void function [-Wreturn-type]
   68 | }
      | ^

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