Submission #303068

#	Time	Username	Problem	Language	Result	Execution time	Memory
303068		tutis	카니발 티켓 (IOI20_tickets)	C++17	25 / 100	1126 ms	62372 KiB

tickets

#include "tickets.h"
#include <vector>
#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<vector<int>> answer(n, vector<int>(m, -1));
	int mxp = n / 2 * k;
	int M[n], P[n];
	vector<pair<int, int>>A;
	for (int i = 0; i < n; i++)
	{
		M[i] = P[i] = 0;
		for (int j = 0; j < m; j++)
			A.push_back({x[i][j], i});
	}
	sort(A.begin(), A.end());
	for (int i = 0; i < n * m / 2; i++)
	{
		M[A[i].second]++;
	}
	reverse(A.begin(), A.end());
	for (int i = 0; i < n * m / 2; i++)
	{
		P[A[i].second]++;
	}
	int cnt = m - k;
	while (cnt > 0)
	{	cnt--;
		pair<long long, bool> mx[n + 1][n / 2 + 2];
		for (int i = 0; i <= n; i++)
			for (int j = 0; j <= n / 2; j++)
				mx[i][j] = { -1e18, false};
		mx[0][0] = {0, false};
		for (int i = 0; i < n; i++)
		{
			if (M[i] >= 1)
				for (int j = 0; j <= n / 2; j++)
				{
					mx[i + 1][j] = max(mx[i + 1][j], {mx[i][j].first + x[i][M[i] - 1], false});
				}
			if (P[i] >= 1)
			{
				for (int j = 0; j <= n / 2; j++)
				{
					mx[i + 1][j + 1] = max(mx[i + 1][j + 1], {mx[i][j].first - x[i][m - P[i]], true});
				}
			}
		}
		int p = n / 2;
		for (int i = n - 1; i >= 0; i--)
		{
			if (mx[i + 1][p].second)
			{
				p--;
				P[i]--;
			}
			else
			{
				M[i]--;
			}
		}
	}
	int a[n];
	int b[n];
	for (int i = 0; i < n; i++)
	{
		a[i] = 0;
		b[i] = m - 1;
	}
	long long ret = 0;
	for (int g = 0; g < k; g++)
	{
		pair<int, int>xx[n];
		for (int i = 0; i < n; i++)
			xx[i] = {P[i], i};
		sort(xx, xx + n);
		for (int i = 0; i < n / 2; i++)
		{
			int id = xx[i].second;
			ret -= x[id][a[id]];
			answer[id][a[id]] = g;
			a[id]++;
			M[id]--;
		}
		for (int i = n / 2; i < n; i++)
		{
			int id = xx[i].second;
			ret += x[id][b[id]];
			answer[id][b[id]] = g;
			b[id]--;
			P[id]--;
		}
	}
	allocate_tickets(answer);
	return ret;
}

Compilation message (stderr)

tickets.cpp: In function 'long long int find_maximum(int, std::vector<std::vector<int> >)':
tickets.cpp:9:6: warning: unused variable 'mxp' [-Wunused-variable]
    9 |  int mxp = n / 2 * k;
      |      ^~~

• 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