Submission #133258

#	Time	Username	Problem	Language	Result	Execution time	Memory
133258		E869120	Olympiads (BOI19_olympiads)	C++14	44 / 100	2029 ms	209248 KiB

olympiads

#include <iostream>
#include <vector>
#include <algorithm>
#include <queue>
#include <functional>
#include <map>
using namespace std;

int N, K, C, A[509][6];

struct Node {
	int a[6];
};

bool operator<(const Node &a1, const Node &a2) {
	for (int i = 0; i < K; i++) {
		if (a1.a[i] < a2.a[i]) return true;
		if (a1.a[i] > a2.a[i]) return false;
	}
	return false;
}

priority_queue<pair<int, Node>>Q;
map<Node, int> Map;

int calc(Node L) {
	int R[6] = { 0, 0, 0, 0, 0, 0 };
	for (int i = 0; i < K; i++) {
		for (int j = 0; j < K; j++) R[j] = max(R[j], A[L.a[i]][j]);
	}
	int sum = 0;
	for (int i = 0; i < K; i++) sum += R[i];
	return sum;
}

int main() {
	cin >> N >> K >> C;
	for (int i = 0; i < N; i++) {
		for (int j = 0; j < K; j++) cin >> A[i][j];
	}

	vector<int>E;
	for (int i = 0; i < K; i++) {
		int maxn = -1, maxid = -1;
		for (int j = 0; j < N; j++) {
			if (maxn < A[j][i]) { maxn = A[j][i]; maxid = j; }
		}
		E.push_back(maxid);
	}
	sort(E.begin(), E.end());
	E.erase(unique(E.begin(), E.end()), E.end());
	for (int i = 0; i < N; i++) {
		if (E.size() == K) break;
		E.push_back(i);
		sort(E.begin(), E.end());
		E.erase(unique(E.begin(), E.end()), E.end());
	}

	Node EE; for (int i = 0; i < K; i++) EE.a[i] = E[i];

	Q.push(make_pair(calc(EE), EE));
	Map[EE] = 1;

	int ret = 0;
	for (int i = 1; i <= C; i++) {
		Node F = Q.top().second;
		ret = Q.top().first; Q.pop();

		vector<int> D;
		for (int j = 0; j < N; j++) {
			bool flag = true;
			for (int k = 0; k < K; k++) {
				if (F.a[k] == j) flag = false;
			}
			if (flag == true) D.push_back(j);
		}

		for (int j = 0; j < K; j++) {
			for (int k : D) {
				Node G = F; G.a[j] = k;
				sort(G.a, G.a + K);
				if (Map[G] == 1) continue;

				Map[G] = 1;
				int Z = calc(G);
				Q.push(make_pair(Z, G));
			}
		}
	}
	cout << ret << endl;
	return 0;
}

Compilation message (stderr)

olympiads.cpp: In function 'int main()':
olympiads.cpp:53:16: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   if (E.size() == K) break;
       ~~~~~~~~~^~~~

• Subtask 1: 13 / 13
• Subtask 2: 31 / 31
• Subtask 3: 0 / 24
• Subtask 4: 0 / 32