Submission #433434

#	Time	Username	Problem	Language	Result	Execution time	Memory
433434		OttoTheDino	Carnival Tickets (IOI20_tickets)	C++17	11 / 100	2 ms	716 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-1) 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) answer[j][m-1-(--used[j])] = i;
            else answer[j][lo[j]++] = i;
        }
    }
	allocate_tickets(answer);
	return ans;
}

Compilation message (stderr)

tickets.cpp: In function 'long long int find_maximum(int, std::vector<std::vector<int> >)':
tickets.cpp:22:44: warning: operation on 'used[c]' may be undefined [-Wsequence-point]
   22 |         if (used[c]<k-1) pq.push({x[c][m-1-++used[c]]+x[c][k-1-used[c]],c});
      |                                            ^~~~~~~~~

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