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 <cmath>
#include <cstdio>
#include <vector>
#include <iostream>
#include <algorithm>
#include <string>
using namespace std;
typedef long long ll;
typedef vector<int> vi;
typedef vector<vi> v2i;
typedef vector<ll> vl;
typedef vector<vl> v2l;
typedef pair<int, int> pi;
typedef vector<pi> vpi;
typedef vector<bool> vb;
#define pb push_back
#define mp make_pair
#define rept(i, a, b) for (int i = a; i < b; i++)
#define rep(i, n) for (int i = 0; i < n; i++)
ll sub1(vi r, vi b) {
int n = r.size(), m = b.size();
v2l dp(n, vl(m));
dp[0][0] = abs(r[0] - b[0]);
rep(i, n) {
rep(j, m) {
if (max(i, j) == 0) continue;
if (i == 0) {
dp[i][j] = dp[i][j - 1] + abs(r[i] - b[j]);
}
else if (j == 0) {
dp[i][j] = dp[i - 1][j] + abs(r[i] - b[j]);
}
else {
dp[i][j] = min({ dp[i - 1][j], dp[i][j - 1], dp[i - 1][j - 1] }) + abs(r[i] - b[j]);
}
}
}
return dp[n - 1][m - 1];
}
long long min_total_length(std::vector<int> r, std::vector<int> b) {
int n = r.size(), m = b.size();
if (max(n, m) < 200) {
return sub1(r, b);
}
ll length = 0;
b.resize(max(m, n), b[0]);
r.resize(max(m, n), r[n - 1]);
rep(i, r.size()) length += b[i] - r[r.size() - i - 1];
return length;
}
Compilation message (stderr)
wiring.cpp: In function 'long long int min_total_length(std::vector<int>, std::vector<int>)':
wiring.cpp:22:37: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
22 | #define rep(i, n) for (int i = 0; i < n; i++)
......
53 | rep(i, r.size()) length += b[i] - r[r.size() - i - 1];
| ~~~~~~~~~~~
wiring.cpp:53:2: note: in expansion of macro 'rep'
53 | rep(i, r.size()) length += b[i] - r[r.size() - i - 1];
| ^~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |