제출 #591621

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
591621		Soumya1	카니발 티켓 (IOI20_tickets)	C++17	100 / 100	766 ms	83976 KiB

tickets

#include "tickets.h"
#include <bits/stdc++.h>
using namespace std;
long long find_maximum(int k, vector<vector<int>> x) {
	int n = x.size();
	int m = x[0].size();
	vector<int> ptr(n);
	priority_queue<pair<int, int>> pq;
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < k; j++) {
			pq.push({-x[i][j] - x[i][m - k + j], i});
		}
	}
	long long ret = 0;
	int taken = 0;
	while (taken < n * k / 2) {
		auto [val, i] = pq.top();
		pq.pop();
		ptr[i]++;
		taken++;
	}
	int cur = 0;
	vector<int> ending(n);
	vector<vector<int>> ans(n, vector<int> (m, -1));
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < ptr[i]; j++) {
			ret -= x[i][j];
			ans[i][j] = cur++;
			if (cur == k) cur = 0;
		}
		ending[i] = cur;
	}
	for (int i = 0; i < n; i++) {
		cur = ending[i];
		for (int j = 0; j < k - ptr[i]; j++) {
			ret += x[i][m - j - 1];
			ans[i][m - j - 1] = cur++;
			if (cur == k) cur = 0;
		}
	}
	allocate_tickets(ans);
	return ret;
}

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