Submission #578266

#	Time	Username	Problem	Language	Result	Execution time	Memory
578266		Vanilla	Carnival Tickets (IOI20_tickets)	C++17	27 / 100	468 ms	51736 KiB

tickets

#include <bits/stdc++.h>
#include "tickets.h"
using namespace std;
 
struct el {
	long long x, y, idx;
 
	bool operator < (const el &oth) const {
		return y + x < oth.x + oth.y;
	}
};

struct val {
	long long v, x, y;

	bool operator < (const val &oth) const {
		return v < oth.v;
	}
};
 
// bool comp (pair <long, long> a, pair <long, long> b) {
// 	if (a.first != b.first) return a.first < 
// }
 
long long find_maximum(int k, vector<vector<int>> x) {
	long long n = x.size();
	long long m = x[0].size();
	long long rs = 0;
	vector <int> times (n, 1);
	vector <vector <int> > answer (n, vector <int> (m, -1));
	vector <vector <bool> > take (n, vector <bool> (m, 0));
	priority_queue <el> pq;
	for (int i = 0; i < n; i++){
		for (int j = 0; j < k; j++){
			rs-=x[i][j];
			take[i][j] = 1;
		}
		pq.push({x[i][k-1], x[i][m-1], i});
	}
	for (int z = 0; z < n * k / 2; z++){
		el now = pq.top();
		pq.pop();
		rs+=now.x + now.y;
		take[now.idx][k - times[now.idx]] = 0;
		take[now.idx][m - times[now.idx]] = 1;
		if (times[now.idx] != k) {
			times[now.idx]++;
			now.x = x[now.idx][k - times[now.idx]];
			now.y = x[now.idx][m - times[now.idx]];
			pq.push(now);
		}
	}
	vector <val> v;
	for (int i = 0; i < n; i++){
		for (int j = 0; j < m; j++){
			if (take[i][j]) {
				v.push_back({x[i][j], i, j});
			}
		}
	}
	sort(v.begin(), v.end());
	for (int z = 0; z < k; z++){
		vector <int> row (n, 0);
		int low = 0, high = 0;
		for (int i = 0; i < n * k && low != n / 2; i++){
			if (!row[v[i].x] && answer[v[i].x][v[i].y] == -1) {
				low++;
				row[v[i].x] = 1;
				answer[v[i].x][v[i].y] = z;
			}
		}
		for (int i = n * k - 1; i >= 0 && high != n / 2; i--){
			if (!row[v[i].x] && answer[v[i].x][v[i].y] == -1) {
				row[v[i].x] = 1;
				answer[v[i].x][v[i].y] = z;
				high++;
			}
		}
	}
	allocate_tickets(answer);
	return rs;
}

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