Submission #931517

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
931517	2024-02-22T01:18:20 Z	beaboss	Carnival Tickets (IOI20_tickets)	C++14	100 / 100	669 ms	76356 KB

tickets

// Source: https://oj.uz/problem/view/IOI20_tickets
// 

#include "tickets.h"

#include "bits/stdc++.h"

using namespace std;

#define s second
#define f first
#define pb push_back

typedef long long ll;

typedef pair<ll, ll> pii;
typedef vector<pii> vpii;

typedef vector<ll> vi;

#define FOR(i, a, b) for (ll i = (a); i<b; i++)

bool ckmin(ll& a, ll b){ return b < a ? a = b, true : false; }

bool ckmax(ll& a, ll b){ return b > a ? a = b, true : false; }

// void allocate_tickets(vector<vi> s) {
// 	// for (auto val: s) {
// 	// 	for (auto j : val) cout << j << ' ';
// 	// 	cout << endl;
// 	// }
// }


ll find_maximum(int k, vector<vector<int> > x) {
	ll n = x.size();
	ll m = x[0].size();
	vi state(n, m - k); // stores range of indices for larger half
	vi start(n, 0);
	ll res = 0;

	priority_queue<pii> q;

	FOR(i, 0, n) {
		FOR(j, m-k, m) res += x[i][j];

		q.push( {-x[i][state[i]] - x[i][start[i]], i});
	}

	FOR(_, 0, k * n / 2) {
		ll cur = q.top().s;
		res += q.top().f;
		q.pop();

		state[cur]++;
		start[cur]++;

		if (state[cur] != m) q.push( {-x[cur][state[cur]] - x[cur][start[cur]], cur});
	}

	vector<vector<int> > final(n, vector<int> (m, -1));

	ll round = 0;

	FOR(i, 0, n) {
		FOR(j, 0, start[i]) {
			final[i][j] = round;
			round= (round + 1) % k;
		}
	}


	FOR(i, 0, n) {
		set<ll> vals;
		FOR(i, 0, k) vals.insert(i);
		for (auto val: final[i]) if (val != -1) vals.erase(val);

		FOR(j, state[i], m) {
			final[i][j] = *vals.begin();
			vals.erase(vals.begin());
		}
	}
	allocate_tickets(final);

	return res;

}




// ll main() {
// 	// ios::sync_with_stdio(false);
// 	// cin.tie(nullptr);

// 	// ll res = find_maximum(2, {{0, 2, 5}, {1, 1, 3}});
// 	// cout << res << endl;

// }

Subtask #1

11.0 / 11.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	0 ms	344 KB	Output is correct
2	Correct	0 ms	348 KB	Output is correct
3	Correct	0 ms	348 KB	Output is correct
4	Correct	0 ms	348 KB	Output is correct
5	Correct	1 ms	348 KB	Output is correct
6	Correct	1 ms	860 KB	Output is correct

Subtask #2

16.0 / 16.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	2 ms	348 KB	Output is correct
2	Correct	0 ms	348 KB	Output is correct
3	Correct	0 ms	348 KB	Output is correct
4	Correct	2 ms	348 KB	Output is correct
5	Correct	15 ms	2332 KB	Output is correct
6	Correct	387 ms	51404 KB	Output is correct

Subtask #3

14.0 / 14.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	0 ms	348 KB	Output is correct
2	Correct	1 ms	352 KB	Output is correct
3	Correct	0 ms	348 KB	Output is correct
4	Correct	2 ms	604 KB	Output is correct
5	Correct	18 ms	2652 KB	Output is correct
6	Correct	446 ms	56916 KB	Output is correct
7	Correct	486 ms	57168 KB	Output is correct
8	Correct	3 ms	604 KB	Output is correct
9	Correct	0 ms	348 KB	Output is correct
10	Correct	0 ms	348 KB	Output is correct
11	Correct	0 ms	348 KB	Output is correct
12	Correct	5 ms	860 KB	Output is correct
13	Correct	16 ms	2140 KB	Output is correct
14	Correct	17 ms	2140 KB	Output is correct
15	Correct	526 ms	57940 KB	Output is correct

Subtask #4

14.0 / 14.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	0 ms	348 KB	Output is correct
2	Correct	0 ms	432 KB	Output is correct
3	Correct	0 ms	348 KB	Output is correct
4	Correct	2 ms	600 KB	Output is correct
5	Correct	25 ms	3412 KB	Output is correct
6	Correct	5 ms	952 KB	Output is correct
7	Correct	5 ms	964 KB	Output is correct
8	Correct	669 ms	76016 KB	Output is correct
9	Correct	622 ms	70920 KB	Output is correct
10	Correct	618 ms	71020 KB	Output is correct
11	Correct	666 ms	76356 KB	Output is correct

Subtask #5

12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	0 ms	344 KB	Output is correct
2	Correct	2 ms	604 KB	Output is correct
3	Correct	2 ms	604 KB	Output is correct
4	Correct	2 ms	604 KB	Output is correct
5	Correct	2 ms	604 KB	Output is correct
6	Correct	2 ms	604 KB	Output is correct
7	Correct	1 ms	348 KB	Output is correct
8	Correct	1 ms	348 KB	Output is correct
9	Correct	2 ms	604 KB	Output is correct
10	Correct	2 ms	604 KB	Output is correct
11	Correct	2 ms	604 KB	Output is correct
12	Correct	2 ms	600 KB	Output is correct

Subtask #6

23.0 / 23.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	0 ms	344 KB	Output is correct
2	Correct	2 ms	604 KB	Output is correct
3	Correct	2 ms	604 KB	Output is correct
4	Correct	2 ms	604 KB	Output is correct
5	Correct	2 ms	604 KB	Output is correct
6	Correct	2 ms	604 KB	Output is correct
7	Correct	1 ms	348 KB	Output is correct
8	Correct	1 ms	348 KB	Output is correct
9	Correct	2 ms	604 KB	Output is correct
10	Correct	2 ms	604 KB	Output is correct
11	Correct	2 ms	604 KB	Output is correct
12	Correct	2 ms	600 KB	Output is correct
13	Correct	17 ms	3164 KB	Output is correct
14	Correct	16 ms	3160 KB	Output is correct
15	Correct	23 ms	3160 KB	Output is correct
16	Correct	26 ms	3416 KB	Output is correct
17	Correct	1 ms	348 KB	Output is correct
18	Correct	2 ms	604 KB	Output is correct
19	Correct	1 ms	348 KB	Output is correct
20	Correct	21 ms	2712 KB	Output is correct
21	Correct	22 ms	3164 KB	Output is correct
22	Correct	24 ms	2608 KB	Output is correct
23	Correct	28 ms	3264 KB	Output is correct

Subtask #7

10.0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	0 ms	344 KB	Output is correct
2	Correct	0 ms	348 KB	Output is correct
3	Correct	0 ms	348 KB	Output is correct
4	Correct	0 ms	348 KB	Output is correct
5	Correct	1 ms	348 KB	Output is correct
6	Correct	1 ms	860 KB	Output is correct
7	Correct	2 ms	348 KB	Output is correct
8	Correct	0 ms	348 KB	Output is correct
9	Correct	0 ms	348 KB	Output is correct
10	Correct	2 ms	348 KB	Output is correct
11	Correct	15 ms	2332 KB	Output is correct
12	Correct	387 ms	51404 KB	Output is correct
13	Correct	0 ms	348 KB	Output is correct
14	Correct	1 ms	352 KB	Output is correct
15	Correct	0 ms	348 KB	Output is correct
16	Correct	2 ms	604 KB	Output is correct
17	Correct	18 ms	2652 KB	Output is correct
18	Correct	446 ms	56916 KB	Output is correct
19	Correct	486 ms	57168 KB	Output is correct
20	Correct	3 ms	604 KB	Output is correct
21	Correct	0 ms	348 KB	Output is correct
22	Correct	0 ms	348 KB	Output is correct
23	Correct	0 ms	348 KB	Output is correct
24	Correct	5 ms	860 KB	Output is correct
25	Correct	16 ms	2140 KB	Output is correct
26	Correct	17 ms	2140 KB	Output is correct
27	Correct	526 ms	57940 KB	Output is correct
28	Correct	0 ms	348 KB	Output is correct
29	Correct	0 ms	432 KB	Output is correct
30	Correct	0 ms	348 KB	Output is correct
31	Correct	2 ms	600 KB	Output is correct
32	Correct	25 ms	3412 KB	Output is correct
33	Correct	5 ms	952 KB	Output is correct
34	Correct	5 ms	964 KB	Output is correct
35	Correct	669 ms	76016 KB	Output is correct
36	Correct	622 ms	70920 KB	Output is correct
37	Correct	618 ms	71020 KB	Output is correct
38	Correct	666 ms	76356 KB	Output is correct
39	Correct	0 ms	344 KB	Output is correct
40	Correct	2 ms	604 KB	Output is correct
41	Correct	2 ms	604 KB	Output is correct
42	Correct	2 ms	604 KB	Output is correct
43	Correct	2 ms	604 KB	Output is correct
44	Correct	2 ms	604 KB	Output is correct
45	Correct	1 ms	348 KB	Output is correct
46	Correct	1 ms	348 KB	Output is correct
47	Correct	2 ms	604 KB	Output is correct
48	Correct	2 ms	604 KB	Output is correct
49	Correct	2 ms	604 KB	Output is correct
50	Correct	2 ms	600 KB	Output is correct
51	Correct	17 ms	3164 KB	Output is correct
52	Correct	16 ms	3160 KB	Output is correct
53	Correct	23 ms	3160 KB	Output is correct
54	Correct	26 ms	3416 KB	Output is correct
55	Correct	1 ms	348 KB	Output is correct
56	Correct	2 ms	604 KB	Output is correct
57	Correct	1 ms	348 KB	Output is correct
58	Correct	21 ms	2712 KB	Output is correct
59	Correct	22 ms	3164 KB	Output is correct
60	Correct	24 ms	2608 KB	Output is correct
61	Correct	28 ms	3264 KB	Output is correct
62	Correct	43 ms	8528 KB	Output is correct
63	Correct	44 ms	8532 KB	Output is correct
64	Correct	72 ms	8532 KB	Output is correct
65	Correct	230 ms	33216 KB	Output is correct
66	Correct	291 ms	33604 KB	Output is correct
67	Correct	5 ms	1112 KB	Output is correct
68	Correct	5 ms	860 KB	Output is correct
69	Correct	392 ms	73224 KB	Output is correct
70	Correct	534 ms	74424 KB	Output is correct
71	Correct	667 ms	75652 KB	Output is correct
72	Correct	584 ms	60752 KB	Output is correct
73	Correct	619 ms	73272 KB	Output is correct
74	Correct	489 ms	59276 KB	Output is correct
75	Correct	516 ms	72276 KB	Output is correct