Submission #747285

#TimeUsernameProblemLanguageResultExecution timeMemory
747285CSQ31Carnival Tickets (IOI20_tickets)C++17
25 / 100
861 ms121440 KiB
#include "tickets.h"
#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 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);}
int mark[2000][2000];
ll pref[2000][2000];
vector<pii>tic[2000];
long long find_maximum(int k, std::vector<std::vector<int>> x) {
	int n = x.size();
	int m = x[0].size();
	vii answer(n);
	ll ans = 0;
	for(int i=0;i<n;i++)answer[i].assign(m,-1);
	vector<array<int,3>>glo;
	for(int i=0;i<n;i++){
		for(int j=0;j<m;j++){
			tic[i].pb({x[i][j],j});
			glo.pb({x[i][j],i,j});
		}
		sort(all(tic[i]));
	}
	sort(all(glo));
	int s = sz(glo);
	for(int i=0;i<n*k/2;i++){
		mark[glo[s-i-1][1]][glo[s-i-1][2]] = 1;
		ans+=glo[s-i-1][0];
		mark[glo[i][1]][glo[i][2]] = -1;
		ans-=glo[i][0];
	}
	
	vector<int>colsum(m,0);
	vii row(n);
	for(int i=0;i<n;i++){
		int z = 0;
		for(int j=0;j<m;j++){
			//cout<<mark[i][j]<<" ";
			if(mark[i][j]){
				colsum[z++]+=mark[i][j];
				row[i].pb(j);
			}
		}
		//cout<<'\n';
	}
	for(int i=0;i<n;i++){
		vector<int>pos,neg;
		for(int j=0;j<sz(row[i]);j++){
			int jj = row[i][j];
			if(mark[i][jj] == -1 && colsum[j] < 0)neg.pb(j);
			if(mark[i][jj] == 1  && colsum[j] > 0)pos.pb(j);
		}
		while(!neg.empty() && !pos.empty()){
			int v = neg.back();
			int u = pos.back();
			neg.pop_back();
			pos.pop_back();
			swap(row[i][v],row[i][u]);
			colsum[v]+=2;
			colsum[u]-=2;
		}
	}
	for(int i=0;i<n;i++){
		for(int j=0;j<sz(row[i]);j++){
				answer[i][row[i][j]] = j;
		}
	}
	if(k==m)allocate_tickets(answer);
	return ans;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...