# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
733115 | t6twotwo | Wiring (IOI17_wiring) | C++17 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include "wiring.h"
#include <bits/stdc++.h>
using namespace std;
constexpr int64_t inf = 1E16;
long long min_total_length(vector<int> R, vector<int> B) {
int N = R.size();
int M = B.size();
vector<array<int, 2>> T;
for (int x : R) {
T.push_back({x, 0});
}
for (int x : B) {
T.push_back({x, 1});
}
sort(T.begin(), T.end());
vector<int64_t> pfs(N + M + 1);
for (int i = 0; i < N + M; i++) {
pfs[i + 1] = pfs[i] + T[i][0];
}
vector<int> d(N + M, -1);
vector<int64_t> dp(N + M + 1, inf);
dp[0] = 0;
for (int i = 0; i < N + M; i++) {
if (i > 0) {
if (T[i][1] == T[i - 1][1]) {
d[i] = d[i - 1];
} else {
d[i] = i - 1;
}
}
if (d[i] == -1) {
continue;
}
int j = d[i];
dp[i + 1] = min(dp[i + 1], dp[j + 1] + (pfs[i + 1] - pfs[j + 1]) - 1LL * (i - j) * T[j][0]);
for (int p = d[j] + 1; p <= j; p++) {
int m = min(j - p + 1, i - j);
int64_t s = (pfs[i + 1] - pfs[i - m + 1]) - (pfs[p + m] - pfs[p]);
if (m <= j - p) {
s += 1LL * T[j + 1][0] * ((j + 1) - (p + m)) - (pfs[j + 1] - pfs[p + m]);
}
if (m < i - j) {
s -= 1LL * T[j][0] * ((i - m + 1) - (j + 1)) - (pfs[i - m + 1] - pfs[j + 1]);
}
dp[i + 1] = min(dp[i + 1], dp[p] + s);
}
}
return dp[N + M];
}