| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 1290059 | vpinx | Nile (IOI24_nile) | C++20 | 0 ms | 0 KiB |
#include "nile.h"
#include <bits/stdc++.h>
using namespace std;
vector<long long> calculate_costs(vector<int> w, vector<int> a, vector<int> b, vector<int> e) {
long long n = w.size(), mx = 0;
vector<pair<int, int>> v(n);
for (int i = 0; i < n; i++) {
v[i] = {w[i], a[i] - b[i]};
mx += a[i];
}
sort(v.begin(), v.end());
auto op = [&](vector<pair<int, int>> &c, int sm) -> long long {
long long sum = 0;
for (int i = 0; i < c.size(); i++) sum += c[i].second;
if ((int)c.size() % 2 == 0) return sum;
long long ans = 0;
for (int i = 0; i < c.size(); i++) {
if (i & 1) {
if (i < c.size() - 1 and c[i + 1].first - c[i - 1].first <= sm) ans = max(ans, sum - c[i].second);
}else ans = max(ans, sum - c[i].second);
return ans;
};
int q = e.size();
vector<long long> ans(q);
for (int t = 0; t < q; t++) {
long long val = 0;
vector<pair<int, int>> cur = {v[0]};
for (int i = 1; i < n; i++) {
if (v[i].first - v[i - 1].first > e[t]) {
val += op(cur, e[t]);
cur.clear();
}
cur.push_back(v[i]);
}
ans[t] = mx - val - op(cur, e[t]);
}
return ans;
}
