Submission #433442

#	Time	Username	Problem	Language	Result	Execution time	Memory
433442		OttoTheDino	Carnival Tickets (IOI20_tickets)	C++17	27 / 100	621 ms	51332 KiB

tickets

#include "tickets.h"
#include <bits/stdc++.h>
using namespace std;

#define rep(i,s,e)                  for (int i = s; i <= e; ++i)
typedef vector<int> vi;

long long find_maximum(int k, vector<vi> x) {
	int n = x.size(), m = x[0].size();
    long long ans = 0;
    vector<vi> answer(n, vi(m, -1));
    priority_queue<pair<int,int>> pq;
    int used[n]={}, lo[n]={};
    rep (i,0,n-1) {
        rep (j,0,k-1) ans -= x[i][j];
        pq.push({x[i][m-1]+x[i][k-1],i});
    }
    rep (i,1,n*k/2) {
        int ma = pq.top().first, c = pq.top().second;
        pq.pop();
        ans += ma;
        if (used[c]<k) {
            ++used[c];
            if (used[c]<k) pq.push({x[c][m-1-used[c]]+x[c][k-1-used[c]],c});
        }
    }
    rep (i,0,k-1) {
        int cnt = 0;
        rep (j,0,n-1) {
            if (used[j] && cnt<n/2) {
                ++cnt;
                answer[j][m-1-(--used[j])] = i;
            }
            else answer[j][lo[j]++] = i;
        }
    }
	allocate_tickets(answer);
	return ans;
}

• 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