#include "wiring.h"
#include <bits/stdc++.h>
using namespace std;
#define forn(i,n) for(int i=0;i<n;++i)
#define pb push_back
#define ll long long
#define pi pair<int,int>
#define f first
#define s second
#define all(x) x.begin(),x.end()
#define int ll
const int inf = 1e18;
int cost(vector<pi>&a, vector<int>&pr, int nw, int i, int j) {
int fcnt = nw - j;
int scnt = i - nw + 1;
int fsum = pr[nw] - pr[j];
int ssum = pr[i+1] - pr[nw];
if (fcnt < scnt) {
fsum += a[nw-1].f * (scnt-fcnt);
}
if (scnt < fcnt) {
ssum += a[nw].f * (fcnt-scnt);
}
return ssum - fsum;
}
int min_total_length(vector<int32_t> r, vector<int32_t> b) {
int n=r.size(),m=b.size();
vector<pi> a;
forn(i,n) a.pb({r[i],0});
forn(i,m) a.pb({b[i],1});
sort(all(a));
vector<int> dp(n+m+1,inf);
dp[0]=0;
vector<int> opt(n+m+1,-1);
int i=0;
while (a[i].s==a[0].s) ++i;
int old=0, nw=i;
vector<int> pr(n+m+1);
forn(i,n+m) pr[i+1]=pr[i]+a[i].f;
dp[i+1] = cost(a,pr,i,i,0);
opt[i+1] = 0;
++i;
for (;i<n+m;++i) {
if (a[i].s != a[i-1].s) {
old=nw;
nw=i;
int j=nw-1;
for (; j>old; --j) {
int x = dp[j]+cost(a,pr,nw,i,j);
int y = dp[j-1]+cost(a,pr,nw,i,j-1);
if (y>x) break;
}
opt[i+1] = j;
dp[i+1] = min(dp[j],dp[j+1])+cost(a,pr,nw,i,j);
} else {
int j=opt[i];
for (; j>old; --j) {
int x = dp[j]+cost(a,pr,nw,i,j);
int y = dp[j-1]+cost(a,pr,nw,i,j-1);
if (y>x) break;
}
opt[i+1] = j;
dp[i+1] = min(dp[j],dp[j+1])+cost(a,pr,nw,i,j);
}
}
return dp[n+m];
}
#undef int
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
212 KB |
Output is correct |
2 |
Correct |
0 ms |
212 KB |
Output is correct |
3 |
Correct |
0 ms |
212 KB |
Output is correct |
4 |
Correct |
1 ms |
212 KB |
Output is correct |
5 |
Correct |
0 ms |
212 KB |
Output is correct |
6 |
Correct |
1 ms |
224 KB |
Output is correct |
7 |
Correct |
0 ms |
212 KB |
Output is correct |
8 |
Correct |
1 ms |
304 KB |
Output is correct |
9 |
Correct |
1 ms |
212 KB |
Output is correct |
10 |
Correct |
0 ms |
212 KB |
Output is correct |
11 |
Correct |
0 ms |
212 KB |
Output is correct |
12 |
Correct |
1 ms |
212 KB |
Output is correct |
13 |
Correct |
0 ms |
212 KB |
Output is correct |
14 |
Correct |
1 ms |
300 KB |
Output is correct |
15 |
Correct |
1 ms |
212 KB |
Output is correct |
16 |
Correct |
1 ms |
264 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
276 KB |
Output is correct |
2 |
Correct |
1 ms |
212 KB |
Output is correct |
3 |
Correct |
27 ms |
7272 KB |
Output is correct |
4 |
Correct |
20 ms |
7292 KB |
Output is correct |
5 |
Correct |
20 ms |
7352 KB |
Output is correct |
6 |
Correct |
27 ms |
9660 KB |
Output is correct |
7 |
Correct |
26 ms |
9676 KB |
Output is correct |
8 |
Correct |
26 ms |
9664 KB |
Output is correct |
9 |
Correct |
26 ms |
9676 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
0 ms |
212 KB |
Output is correct |
3 |
Correct |
39 ms |
9644 KB |
Output is correct |
4 |
Correct |
40 ms |
9732 KB |
Output is correct |
5 |
Correct |
1 ms |
212 KB |
Output is correct |
6 |
Correct |
0 ms |
212 KB |
Output is correct |
7 |
Correct |
0 ms |
212 KB |
Output is correct |
8 |
Correct |
0 ms |
212 KB |
Output is correct |
9 |
Correct |
0 ms |
212 KB |
Output is correct |
10 |
Correct |
0 ms |
300 KB |
Output is correct |
11 |
Correct |
1 ms |
212 KB |
Output is correct |
12 |
Correct |
0 ms |
296 KB |
Output is correct |
13 |
Correct |
1 ms |
212 KB |
Output is correct |
14 |
Correct |
1 ms |
212 KB |
Output is correct |
15 |
Correct |
1 ms |
212 KB |
Output is correct |
16 |
Correct |
1 ms |
212 KB |
Output is correct |
17 |
Correct |
1 ms |
212 KB |
Output is correct |
18 |
Correct |
38 ms |
9732 KB |
Output is correct |
19 |
Correct |
40 ms |
9696 KB |
Output is correct |
20 |
Correct |
41 ms |
9740 KB |
Output is correct |
21 |
Correct |
38 ms |
9716 KB |
Output is correct |
22 |
Correct |
38 ms |
9772 KB |
Output is correct |
23 |
Correct |
40 ms |
9780 KB |
Output is correct |
24 |
Correct |
45 ms |
9768 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
33 ms |
9656 KB |
Output is correct |
3 |
Correct |
36 ms |
9648 KB |
Output is correct |
4 |
Correct |
36 ms |
9632 KB |
Output is correct |
5 |
Correct |
36 ms |
9596 KB |
Output is correct |
6 |
Correct |
0 ms |
212 KB |
Output is correct |
7 |
Correct |
0 ms |
212 KB |
Output is correct |
8 |
Correct |
0 ms |
212 KB |
Output is correct |
9 |
Correct |
0 ms |
212 KB |
Output is correct |
10 |
Correct |
0 ms |
212 KB |
Output is correct |
11 |
Correct |
0 ms |
212 KB |
Output is correct |
12 |
Correct |
1 ms |
212 KB |
Output is correct |
13 |
Correct |
1 ms |
300 KB |
Output is correct |
14 |
Correct |
1 ms |
212 KB |
Output is correct |
15 |
Correct |
1 ms |
212 KB |
Output is correct |
16 |
Correct |
0 ms |
212 KB |
Output is correct |
17 |
Correct |
0 ms |
212 KB |
Output is correct |
18 |
Correct |
33 ms |
9784 KB |
Output is correct |
19 |
Correct |
35 ms |
9788 KB |
Output is correct |
20 |
Correct |
33 ms |
9740 KB |
Output is correct |
21 |
Correct |
31 ms |
9784 KB |
Output is correct |
22 |
Correct |
33 ms |
9740 KB |
Output is correct |
23 |
Correct |
28 ms |
9736 KB |
Output is correct |
24 |
Correct |
32 ms |
9768 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
212 KB |
Output is correct |
2 |
Correct |
0 ms |
212 KB |
Output is correct |
3 |
Correct |
0 ms |
212 KB |
Output is correct |
4 |
Correct |
1 ms |
212 KB |
Output is correct |
5 |
Correct |
0 ms |
212 KB |
Output is correct |
6 |
Correct |
1 ms |
224 KB |
Output is correct |
7 |
Correct |
0 ms |
212 KB |
Output is correct |
8 |
Correct |
1 ms |
304 KB |
Output is correct |
9 |
Correct |
1 ms |
212 KB |
Output is correct |
10 |
Correct |
0 ms |
212 KB |
Output is correct |
11 |
Correct |
0 ms |
212 KB |
Output is correct |
12 |
Correct |
1 ms |
212 KB |
Output is correct |
13 |
Correct |
0 ms |
212 KB |
Output is correct |
14 |
Correct |
1 ms |
300 KB |
Output is correct |
15 |
Correct |
1 ms |
212 KB |
Output is correct |
16 |
Correct |
1 ms |
264 KB |
Output is correct |
17 |
Correct |
0 ms |
276 KB |
Output is correct |
18 |
Correct |
1 ms |
212 KB |
Output is correct |
19 |
Correct |
27 ms |
7272 KB |
Output is correct |
20 |
Correct |
20 ms |
7292 KB |
Output is correct |
21 |
Correct |
20 ms |
7352 KB |
Output is correct |
22 |
Correct |
27 ms |
9660 KB |
Output is correct |
23 |
Correct |
26 ms |
9676 KB |
Output is correct |
24 |
Correct |
26 ms |
9664 KB |
Output is correct |
25 |
Correct |
26 ms |
9676 KB |
Output is correct |
26 |
Correct |
0 ms |
212 KB |
Output is correct |
27 |
Correct |
0 ms |
212 KB |
Output is correct |
28 |
Correct |
39 ms |
9644 KB |
Output is correct |
29 |
Correct |
40 ms |
9732 KB |
Output is correct |
30 |
Correct |
1 ms |
212 KB |
Output is correct |
31 |
Correct |
0 ms |
212 KB |
Output is correct |
32 |
Correct |
0 ms |
212 KB |
Output is correct |
33 |
Correct |
0 ms |
212 KB |
Output is correct |
34 |
Correct |
0 ms |
212 KB |
Output is correct |
35 |
Correct |
0 ms |
300 KB |
Output is correct |
36 |
Correct |
1 ms |
212 KB |
Output is correct |
37 |
Correct |
0 ms |
296 KB |
Output is correct |
38 |
Correct |
1 ms |
212 KB |
Output is correct |
39 |
Correct |
1 ms |
212 KB |
Output is correct |
40 |
Correct |
1 ms |
212 KB |
Output is correct |
41 |
Correct |
1 ms |
212 KB |
Output is correct |
42 |
Correct |
1 ms |
212 KB |
Output is correct |
43 |
Correct |
38 ms |
9732 KB |
Output is correct |
44 |
Correct |
40 ms |
9696 KB |
Output is correct |
45 |
Correct |
41 ms |
9740 KB |
Output is correct |
46 |
Correct |
38 ms |
9716 KB |
Output is correct |
47 |
Correct |
38 ms |
9772 KB |
Output is correct |
48 |
Correct |
40 ms |
9780 KB |
Output is correct |
49 |
Correct |
45 ms |
9768 KB |
Output is correct |
50 |
Correct |
0 ms |
212 KB |
Output is correct |
51 |
Correct |
33 ms |
9656 KB |
Output is correct |
52 |
Correct |
36 ms |
9648 KB |
Output is correct |
53 |
Correct |
36 ms |
9632 KB |
Output is correct |
54 |
Correct |
36 ms |
9596 KB |
Output is correct |
55 |
Correct |
0 ms |
212 KB |
Output is correct |
56 |
Correct |
0 ms |
212 KB |
Output is correct |
57 |
Correct |
0 ms |
212 KB |
Output is correct |
58 |
Correct |
0 ms |
212 KB |
Output is correct |
59 |
Correct |
0 ms |
212 KB |
Output is correct |
60 |
Correct |
0 ms |
212 KB |
Output is correct |
61 |
Correct |
1 ms |
212 KB |
Output is correct |
62 |
Correct |
1 ms |
300 KB |
Output is correct |
63 |
Correct |
1 ms |
212 KB |
Output is correct |
64 |
Correct |
1 ms |
212 KB |
Output is correct |
65 |
Correct |
0 ms |
212 KB |
Output is correct |
66 |
Correct |
0 ms |
212 KB |
Output is correct |
67 |
Correct |
33 ms |
9784 KB |
Output is correct |
68 |
Correct |
35 ms |
9788 KB |
Output is correct |
69 |
Correct |
33 ms |
9740 KB |
Output is correct |
70 |
Correct |
31 ms |
9784 KB |
Output is correct |
71 |
Correct |
33 ms |
9740 KB |
Output is correct |
72 |
Correct |
28 ms |
9736 KB |
Output is correct |
73 |
Correct |
32 ms |
9768 KB |
Output is correct |
74 |
Correct |
28 ms |
9772 KB |
Output is correct |
75 |
Correct |
35 ms |
9756 KB |
Output is correct |
76 |
Correct |
35 ms |
9804 KB |
Output is correct |
77 |
Correct |
33 ms |
9764 KB |
Output is correct |
78 |
Correct |
31 ms |
9736 KB |
Output is correct |
79 |
Correct |
34 ms |
9752 KB |
Output is correct |
80 |
Correct |
28 ms |
9744 KB |
Output is correct |
81 |
Correct |
34 ms |
9796 KB |
Output is correct |
82 |
Correct |
27 ms |
9756 KB |
Output is correct |
83 |
Correct |
27 ms |
9784 KB |
Output is correct |
84 |
Correct |
34 ms |
9712 KB |
Output is correct |
85 |
Correct |
35 ms |
9712 KB |
Output is correct |
86 |
Correct |
29 ms |
9736 KB |
Output is correct |
87 |
Correct |
30 ms |
9772 KB |
Output is correct |
88 |
Correct |
31 ms |
9740 KB |
Output is correct |
89 |
Correct |
30 ms |
9756 KB |
Output is correct |
90 |
Correct |
45 ms |
9780 KB |
Output is correct |
91 |
Correct |
29 ms |
9772 KB |
Output is correct |
92 |
Correct |
32 ms |
9772 KB |
Output is correct |
93 |
Correct |
30 ms |
9732 KB |
Output is correct |
94 |
Correct |
32 ms |
9812 KB |
Output is correct |
95 |
Correct |
34 ms |
9788 KB |
Output is correct |
96 |
Correct |
27 ms |
9768 KB |
Output is correct |
97 |
Correct |
28 ms |
9784 KB |
Output is correct |
98 |
Correct |
35 ms |
9776 KB |
Output is correct |
99 |
Correct |
30 ms |
9876 KB |
Output is correct |
100 |
Correct |
40 ms |
9716 KB |
Output is correct |
101 |
Correct |
30 ms |
9752 KB |
Output is correct |
102 |
Correct |
37 ms |
9744 KB |
Output is correct |