# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
578774 | thiago_bastos | 카니발 티켓 (IOI20_tickets) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "tickets.h"
#include <vector>
#include <algorithm>
#include <utility>
long long find_maximum(int k, std::vector<std::vector<int>> x) {
int n = x.size();
int m = x[0].size();
long long prize = 0;
auto answer = std :: vector<std :: vector<int>>(n, std :: vector<int>(m, -1));
std :: vector<std :: pair<int, int>> a;
std :: vector<int> l(n, m), r(n, 0), lo(n, 0), hi(n, m - 1);
for(int i = 0; i < n; ++i)
for(int v : x[i])
a.emplace_back(v, i);
std :: sort(a.begin(), a.end());
for(int i = size(a) / 2; i < (int)size(a); ++i) {
auto [v, k] = a[i];
++r[k], --l[k];
}
for(int j = 0; j < k; ++j) {
int L = 0, R = 0;
for(int i = 0; i < n; ++i) {
if(l[i] && !r[i]) {
++L, --l[i];
prize -= x[i][lo[i]];
answer[i][lo[i]++] = j;
} else if(!l[i] && r[i]) {
++R, --r[i];
prize += x[i][hi[i]];
answer[i][hi[i]--] = j;
}
}
vector<std :: pair<int, int>> a;
for(int i = 0; i < n; ++i) {
if(l[i] && r[i]) {
prize += x[i][hi[i]];
answer[i][hi[i]] = j;
a.emplace_back(-x[i][lo[i]] - x[i][hi[i]], i);
--hi[i], --r[i];
}
}
std :: sort(a.rbegin(), a.rend());
for(int i = 0; i < n / 2 - L; ++i) {
auto [v, k] = a[i];
prize += v;
answer[k][hi[k]++] = -1;
answer[k][lo[k]++] = j;
++r[k], --l[k];
}
}
allocate_tickets(answer);
return prize;
}