Submission #779405

#	Time	Username	Problem	Language	Result	Execution time	Memory
779405		Josia	Carnival Tickets (IOI20_tickets)	C++17	11 / 100	1 ms	724 KiB

tickets

#include "tickets.h"
#include <vector>
#include <bits/stdc++.h>

using namespace std;

#define int long long

int find_maximum(signed k, std::vector<std::vector<signed>> x) {
	
	int colors = x.size();
	int ticketsPerColor = x[0].size();

	int rounds = k;


	if (ticketsPerColor == 1) {
		int m = ticketsPerColor;
		int n = colors;
		
		int res = 0;

		assert(m == 1);

		vector<int> vals;
		
		for (int i = 0; i<n; i++) {
			vals.push_back(x[i][0]);
		}

		sort(vals.begin(), vals.end());

		int med = vals[n/2];


		for (int i: vals) res += abs(i-med);
		

		vector<vector<signed>> cards(n, vector<signed>(m));

		for (int i=0; i<n; i++) {
			cards[i][0] = 0;
		}


		allocate_tickets(cards);
		return res;
	}


	vector<array<int, 4>> cards(colors);

	for (int i=0; i<colors; i++) {
		cards[i][0] = (*max_element(x[i].begin(), x[i].end()))-(*min_element(x[i].begin(), x[i].end()));
		cards[i][1] = min_element(x[i].begin(), x[i].end())-x[i].begin();
		cards[i][2] = max_element(x[i].begin(), x[i].end())-x[i].begin();
		cards[i][3] = i;
	}
	sort(cards.begin(), cards.end());

	int res = 0;

	vector<vector<signed>> allocation(colors, vector<signed>(ticketsPerColor, -1));

	for (int i = 0; i<colors/2; i++) {
		res += x[cards[i][3]][cards[i][2]];
		allocation[cards[i][3]][cards[i][2]] = 0;
	}
	for (int i = colors/2; i<colors; i++) {
		res -= x[cards[i][3]][cards[i][1]];
		allocation[cards[i][3]][cards[i][1]] = 0;
	}
	allocate_tickets(allocation);
	return res;
}

Compilation message (stderr)

tickets.cpp: In function 'long long int find_maximum(int, std::vector<std::vector<int> >)':
tickets.cpp:14:6: warning: unused variable 'rounds' [-Wunused-variable]
   14 |  int rounds = k;
      |      ^~~~~~

• 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