#include "tickets.h"
#include <bits/stdc++.h>
using namespace std;
#define FOR(i, j, k, l) for(int i = (j); i < (k); i += (l))
#define FORD(i, j, k, l) for(int i = (j); i >= (k); i -= (l))
#define REP(i, n) FOR(i, 0, n, 1)
#define REPD(i, n) FORD(i, n, 0, 1)
typedef long long ll;
#define pb push_back
long long find_maximum(int k, vector<vector<int>> x) {
int n = x.size();
int m = x[0].size();
// think in terms of change;
vector<array<int, 3>> change;
ll sum = 0;
REP(i, n){
REP(j, k){
change.pb({x[i][j] + x[i][m - k + j], i, j});
sum -= x[i][j];
}
}
sort(change.begin(), change.end(), greater<array<int, 3>>());
vector<int> upper(n, k);
vector<vector<array<int, 2>>> rows(n);
REP(i, n * k / 2){
auto v = change[i];
sum += v[0];
upper[v[1]] = v[2];
}
REP(i, n){
REP(j, upper[i]){
rows[i].pb({x[i][j], j});
}
}
REP(i, n){
REP(j, k - upper[i]){
rows[i].pb({x[i][m - 1 - j], m - 1 - j});
}
}
vector<int> cnts = upper; // cnt smaller available
vector<vector<int>> answer(n, vector<int>(m, -1));
vector<int> sorted(n);
iota(sorted.begin(), sorted.end(), 0);
auto cmp = [&](int f, int s){
return cnts[f] > cnts[s];
};
REP(i, k){
sort(sorted.begin(), sorted.end(), cmp);
REP(j, n / 2){
cnts[sorted[j]]--;
int row = sorted[j]; int col = rows[row][cnts[row]][1];
answer[row][col] = i;
}
FOR(j, n / 2, n, 1){
int row = sorted[j];
int smaller_taken = upper[row] - cnts[row];
int col = k - 1 - i + smaller_taken;
col = rows[row][col][1];
answer[row][col] = i;
}
}
allocate_tickets(answer);
return sum;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
204 KB |
Output is correct |
2 |
Correct |
1 ms |
292 KB |
Output is correct |
3 |
Correct |
1 ms |
204 KB |
Output is correct |
4 |
Correct |
1 ms |
204 KB |
Output is correct |
5 |
Correct |
1 ms |
332 KB |
Output is correct |
6 |
Correct |
3 ms |
716 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
204 KB |
Output is correct |
2 |
Correct |
1 ms |
204 KB |
Output is correct |
3 |
Correct |
1 ms |
288 KB |
Output is correct |
4 |
Correct |
2 ms |
460 KB |
Output is correct |
5 |
Correct |
26 ms |
3244 KB |
Output is correct |
6 |
Correct |
620 ms |
53096 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
204 KB |
Output is correct |
2 |
Correct |
1 ms |
204 KB |
Output is correct |
3 |
Correct |
1 ms |
204 KB |
Output is correct |
4 |
Correct |
3 ms |
452 KB |
Output is correct |
5 |
Correct |
33 ms |
3392 KB |
Output is correct |
6 |
Correct |
865 ms |
70712 KB |
Output is correct |
7 |
Correct |
875 ms |
75116 KB |
Output is correct |
8 |
Correct |
5 ms |
660 KB |
Output is correct |
9 |
Correct |
1 ms |
204 KB |
Output is correct |
10 |
Correct |
1 ms |
204 KB |
Output is correct |
11 |
Correct |
1 ms |
204 KB |
Output is correct |
12 |
Correct |
8 ms |
1040 KB |
Output is correct |
13 |
Correct |
33 ms |
3004 KB |
Output is correct |
14 |
Correct |
29 ms |
3140 KB |
Output is correct |
15 |
Correct |
991 ms |
87900 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
292 KB |
Output is correct |
2 |
Correct |
1 ms |
204 KB |
Output is correct |
3 |
Correct |
1 ms |
204 KB |
Output is correct |
4 |
Correct |
3 ms |
556 KB |
Output is correct |
5 |
Correct |
42 ms |
5072 KB |
Output is correct |
6 |
Correct |
7 ms |
1036 KB |
Output is correct |
7 |
Correct |
8 ms |
1392 KB |
Output is correct |
8 |
Correct |
1181 ms |
96828 KB |
Output is correct |
9 |
Correct |
1107 ms |
92912 KB |
Output is correct |
10 |
Correct |
1052 ms |
91920 KB |
Output is correct |
11 |
Correct |
1220 ms |
96748 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
204 KB |
Output is correct |
2 |
Correct |
2 ms |
428 KB |
Output is correct |
3 |
Correct |
3 ms |
460 KB |
Output is correct |
4 |
Correct |
3 ms |
460 KB |
Output is correct |
5 |
Correct |
3 ms |
588 KB |
Output is correct |
6 |
Correct |
3 ms |
588 KB |
Output is correct |
7 |
Correct |
1 ms |
332 KB |
Output is correct |
8 |
Correct |
1 ms |
332 KB |
Output is correct |
9 |
Correct |
3 ms |
460 KB |
Output is correct |
10 |
Correct |
3 ms |
588 KB |
Output is correct |
11 |
Correct |
4 ms |
588 KB |
Output is correct |
12 |
Correct |
3 ms |
588 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
204 KB |
Output is correct |
2 |
Correct |
2 ms |
428 KB |
Output is correct |
3 |
Correct |
3 ms |
460 KB |
Output is correct |
4 |
Correct |
3 ms |
460 KB |
Output is correct |
5 |
Correct |
3 ms |
588 KB |
Output is correct |
6 |
Correct |
3 ms |
588 KB |
Output is correct |
7 |
Correct |
1 ms |
332 KB |
Output is correct |
8 |
Correct |
1 ms |
332 KB |
Output is correct |
9 |
Correct |
3 ms |
460 KB |
Output is correct |
10 |
Correct |
3 ms |
588 KB |
Output is correct |
11 |
Correct |
4 ms |
588 KB |
Output is correct |
12 |
Correct |
3 ms |
588 KB |
Output is correct |
13 |
Correct |
28 ms |
3148 KB |
Output is correct |
14 |
Correct |
26 ms |
3192 KB |
Output is correct |
15 |
Correct |
33 ms |
4028 KB |
Output is correct |
16 |
Correct |
53 ms |
5148 KB |
Output is correct |
17 |
Correct |
2 ms |
332 KB |
Output is correct |
18 |
Correct |
3 ms |
588 KB |
Output is correct |
19 |
Correct |
2 ms |
460 KB |
Output is correct |
20 |
Correct |
32 ms |
3624 KB |
Output is correct |
21 |
Correct |
34 ms |
4120 KB |
Output is correct |
22 |
Correct |
37 ms |
4492 KB |
Output is correct |
23 |
Correct |
40 ms |
4872 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
204 KB |
Output is correct |
2 |
Correct |
1 ms |
292 KB |
Output is correct |
3 |
Correct |
1 ms |
204 KB |
Output is correct |
4 |
Correct |
1 ms |
204 KB |
Output is correct |
5 |
Correct |
1 ms |
332 KB |
Output is correct |
6 |
Correct |
3 ms |
716 KB |
Output is correct |
7 |
Correct |
1 ms |
204 KB |
Output is correct |
8 |
Correct |
1 ms |
204 KB |
Output is correct |
9 |
Correct |
1 ms |
288 KB |
Output is correct |
10 |
Correct |
2 ms |
460 KB |
Output is correct |
11 |
Correct |
26 ms |
3244 KB |
Output is correct |
12 |
Correct |
620 ms |
53096 KB |
Output is correct |
13 |
Correct |
1 ms |
204 KB |
Output is correct |
14 |
Correct |
1 ms |
204 KB |
Output is correct |
15 |
Correct |
1 ms |
204 KB |
Output is correct |
16 |
Correct |
3 ms |
452 KB |
Output is correct |
17 |
Correct |
33 ms |
3392 KB |
Output is correct |
18 |
Correct |
865 ms |
70712 KB |
Output is correct |
19 |
Correct |
875 ms |
75116 KB |
Output is correct |
20 |
Correct |
5 ms |
660 KB |
Output is correct |
21 |
Correct |
1 ms |
204 KB |
Output is correct |
22 |
Correct |
1 ms |
204 KB |
Output is correct |
23 |
Correct |
1 ms |
204 KB |
Output is correct |
24 |
Correct |
8 ms |
1040 KB |
Output is correct |
25 |
Correct |
33 ms |
3004 KB |
Output is correct |
26 |
Correct |
29 ms |
3140 KB |
Output is correct |
27 |
Correct |
991 ms |
87900 KB |
Output is correct |
28 |
Correct |
1 ms |
292 KB |
Output is correct |
29 |
Correct |
1 ms |
204 KB |
Output is correct |
30 |
Correct |
1 ms |
204 KB |
Output is correct |
31 |
Correct |
3 ms |
556 KB |
Output is correct |
32 |
Correct |
42 ms |
5072 KB |
Output is correct |
33 |
Correct |
7 ms |
1036 KB |
Output is correct |
34 |
Correct |
8 ms |
1392 KB |
Output is correct |
35 |
Correct |
1181 ms |
96828 KB |
Output is correct |
36 |
Correct |
1107 ms |
92912 KB |
Output is correct |
37 |
Correct |
1052 ms |
91920 KB |
Output is correct |
38 |
Correct |
1220 ms |
96748 KB |
Output is correct |
39 |
Correct |
1 ms |
204 KB |
Output is correct |
40 |
Correct |
2 ms |
428 KB |
Output is correct |
41 |
Correct |
3 ms |
460 KB |
Output is correct |
42 |
Correct |
3 ms |
460 KB |
Output is correct |
43 |
Correct |
3 ms |
588 KB |
Output is correct |
44 |
Correct |
3 ms |
588 KB |
Output is correct |
45 |
Correct |
1 ms |
332 KB |
Output is correct |
46 |
Correct |
1 ms |
332 KB |
Output is correct |
47 |
Correct |
3 ms |
460 KB |
Output is correct |
48 |
Correct |
3 ms |
588 KB |
Output is correct |
49 |
Correct |
4 ms |
588 KB |
Output is correct |
50 |
Correct |
3 ms |
588 KB |
Output is correct |
51 |
Correct |
28 ms |
3148 KB |
Output is correct |
52 |
Correct |
26 ms |
3192 KB |
Output is correct |
53 |
Correct |
33 ms |
4028 KB |
Output is correct |
54 |
Correct |
53 ms |
5148 KB |
Output is correct |
55 |
Correct |
2 ms |
332 KB |
Output is correct |
56 |
Correct |
3 ms |
588 KB |
Output is correct |
57 |
Correct |
2 ms |
460 KB |
Output is correct |
58 |
Correct |
32 ms |
3624 KB |
Output is correct |
59 |
Correct |
34 ms |
4120 KB |
Output is correct |
60 |
Correct |
37 ms |
4492 KB |
Output is correct |
61 |
Correct |
40 ms |
4872 KB |
Output is correct |
62 |
Correct |
71 ms |
6836 KB |
Output is correct |
63 |
Correct |
75 ms |
6876 KB |
Output is correct |
64 |
Correct |
117 ms |
10908 KB |
Output is correct |
65 |
Correct |
387 ms |
31680 KB |
Output is correct |
66 |
Correct |
511 ms |
41384 KB |
Output is correct |
67 |
Correct |
8 ms |
1292 KB |
Output is correct |
68 |
Correct |
7 ms |
992 KB |
Output is correct |
69 |
Correct |
652 ms |
52460 KB |
Output is correct |
70 |
Correct |
889 ms |
69864 KB |
Output is correct |
71 |
Correct |
1179 ms |
94840 KB |
Output is correct |
72 |
Correct |
1084 ms |
91228 KB |
Output is correct |
73 |
Correct |
1087 ms |
89472 KB |
Output is correct |
74 |
Correct |
869 ms |
69700 KB |
Output is correct |
75 |
Correct |
852 ms |
68804 KB |
Output is correct |