Submission #1050596

#	Time	Username	Problem	Language	Result	Execution time	Memory
1050596		fuad27	Carnival Tickets (IOI20_tickets)	C++17	100 / 100	409 ms	70880 KiB

tickets

#include "tickets.h"
#include <bits/stdc++.h>
using namespace std;
long long find_maximum(int k, std::vector<std::vector<int>> x) {
#define int long long
	int n = x.size();
	int m = x[0].size();
	vector<pair<int,int>> s(n);
	vector<int> all, cnt(n);
	for(int i = 0;i<n;i++) {
		s[i] = pair{0, m-1};
	}
	vector<vector<int32_t>> res(n, vector<int32_t>(m, -1));
 
	priority_queue<pair<int,int>> pq;
	for(int i = 0;i<n;i++) {
			pq.push({-x[i][m-k]-x[i][cnt[i]], i});
	}
	for(int _ = 0;_<(n*k)/2;_++) {
		int i = pq.top().second;
		pq.pop();
		cnt[i]++;
		if(m-k+cnt[i] < m) {
			pq.push({{-x[i][m-k+cnt[i]]-x[i][cnt[i]]}, i});
		}
	}
	long long sum=0;
	for(int _ = 0;_<k;_++) {
		vector<int> idxs(n);
		iota(idxs.begin(), idxs.end(), 0);
		sort(idxs.begin(), idxs.end(), [&](int a, int b) -> bool {
			return cnt[a] < cnt[b];
		});
		vector<int> ans;
		for(int i = 0;i<n/2;i++) {
			sum+=x[idxs[i]][s[idxs[i]].second];
			res[idxs[i]][s[idxs[i]].second--]=_;
		}
		for(int i = n/2;i<n;i++) {
//			assert(cnt[idxs[i]]);
			sum-=x[idxs[i]][s[idxs[i]].first];
			cnt[idxs[i]]--;
			res[idxs[i]][s[idxs[i]].first++]=_;
		}
	}
	allocate_tickets(res);
#undef int
	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