# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
1112542 | SalihSahin | Wiring (IOI17_wiring) | C++14 | 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 <bits/stdc++.h>
#define pb push_back
#define int long long
using namespace std;
#include "wiring.h"
const int inf = 1e17;
long long min_total_length(vector<int> r, vector<int> b) {
int n = r.size(), m = b.size();
vector<int> mndisr(n), mndisb(m);
int indb = 0, indr = 0;
for(int i = 0; i < n; i++){
while(indb + 1 < m && b[indb+1] < r[i]) indb++;
mndisr[i] = min(abs(r[i] - b[indb]), (indb + 1 < m ? abs(r[i] - b[indb+1]) : inf));
}
for(int i = 0; i < m; i++){
while(indr + 1 < n && r[indr+1] < b[i]) indr++;
mndisb[i] = min(abs(b[i] - r[indr]), (indr + 1 < n ? abs(b[i] - r[indr+1]) : inf));
}
vector<vector<int> > dp(n+1, vector<int>(m+1, inf));
dp[0][0] = 0;
for(int i = 0; i <= n; i++){
for(int j = 0; j <= m; j++){
if(i < n) dp[i+1][j] = min(dp[i+1][j], dp[i][j] + mndisr[i]);
if(j < m) dp[i][j+1] = min(dp[i][j+1], dp[i][j] + mndisb[j]);
if(i < n && j < m) dp[i+1][j+1] = min(dp[i+1][j+1], dp[i][j] + abs(r[i] - b[j]));
}
}
return dp[n][m];
}