Submission #1294623

#	Time	Username	Problem	Language	Result	Execution time	Memory
1294623		daotuankhoi	Carnival Tickets (IOI20_tickets)	C++20	100 / 100	472 ms	93288 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<vector<int>> answer(n, vector<int>(m, -1));

	vector<vector<bool>> mark(n, vector<bool>(k));

	vector<pair<long long, int>> val;
	vector<int> big(n, k), small(n, 0);
	long long sum = 0;
	for (int i = 0; i < n; i++) {
        for (int j = m - k; j < m; j++) {
            sum += x[i][j];
            val.emplace_back(x[i][j] + x[i][j - m + k], i);
        }
	}
	sort(val.begin(), val.end());
	for (int i = 0; i < ((n * k) >> 1); i++) {
        sum -= val[i].first;
        big[val[i].second]--;
        small[val[i].second]++;
	}
    int gr = 0;
	for (int i = 0; i < n; i++) {
        for (int j = 0; j < small[i]; j++) {
            mark[i][answer[i][j] = gr] = 1;
            gr++;
            if (gr >= k) gr = 0;
        }
    }
    for (int i = 0; i < n; i++) {
        gr = 0;
        for (int j = m - big[i]; j < m; j++) {
            while (mark[i][gr]) gr++;
            answer[i][j] = gr++;
        }
    }
	allocate_tickets(answer);
	return sum;
}

• 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