제출 #307060

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
307060		rqi	카니발 티켓 (IOI20_tickets)	C++14	25 / 100	1394 ms	76408 KiB

tickets

#include "tickets.h"
#include <bits/stdc++.h>

using namespace std;

typedef pair<int, int> pi;
typedef vector<int> vi;
typedef long long ll;
typedef vector<ll> vl;
typedef pair<ll, ll> pl;
typedef vector<pi> vpi;
typedef vector<pl> vpl;

#define pb push_back
#define sz(x) (int)(x).size()
#define all(x) begin(x), end(x)
#define mp make_pair
#define f first
#define s second
#define bk back()

vi goods[1505];
vi bads[1505];
priority_queue<pi> szs; //size, index

long long find_maximum(int k, vector<vi> x) {
	int n = sz(x);
	int m = sz(x[0]);
	vector<vi> answer;
	for(int i = 0; i < n; i++){
		answer.pb(vi(m, -1));
	}
	ll ans = 0;
	if(k == m){

		vector<pair<int, pi>> inds;
		for(int i = 0; i < n; i++){
			for(int j = 0; j < m; j++){
				inds.pb(mp(x[i][j], mp(i, j)));
				ans-=x[i][j];
			}
		}
		sort(all(inds));
		for(int i = 0; i < (n/2)*m; i++){
			bads[inds[i].s.f].pb(inds[i].s.s);
		}
		for(int i = (n/2)*m; i < n*m; i++){
			goods[inds[i].s.f].pb(inds[i].s.s);
			ans+=2*inds[i].f;
		}
		for(int i = 0; i < n; i++){
			szs.push(mp(sz(goods[i]), i));
		}

		for(int round = 0; round < k; round++){
			vpi trash;
			for(int i = 0; i < n/2; i++){
				pi a = szs.top();
				szs.pop();
				answer[a.s][goods[a.s].bk] = round;
				goods[a.s].pop_back();
				trash.pb(mp(a.f-1, a.s));
			}

			for(int i = 0; i < n/2; i++){
				pi a = szs.top();
				szs.pop();
				//cout << a.s << "\n";
				answer[a.s][bads[a.s].bk] = round;
				bads[a.s].pop_back();
				trash.pb(a);
			}

			for(auto u: trash){
				szs.push(u);
			}
		}
	}
	allocate_tickets(answer);
	return ans;
}

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