#include <bits/stdc++.h>
#include "wiring.h"
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
#define sep ' '
#define X first
#define Y second
#define lc id << 1
#define rc lc | 1
#define all(x) (x).begin() , (x).end()
#define SZ(x) int((x).size())
const int MAXN = 2e5 + 10;
const int MOD = 1e9 + 7;
const ll INF = 1e18;
ll dp[MAXN] , seg[2][MAXN << 2];
void modify(int ind , int x , ll val , int id = 1 , int l = 0 , int r = MAXN){
if(r - l == 1){
seg[ind][id] = val;
return;
}
int mid = l + r >> 1;
if(x < mid) modify(ind , x , val , lc , l , mid);
else modify(ind , x , val , rc , mid , r);
seg[ind][id] = min(seg[ind][lc] , seg[ind][rc]);
}
ll get(int ind , int ql , int qr , int id = 1 , int l = 0 , int r = MAXN){
if(qr <= l || r <= ql) return INF;
if(ql <= l && r <= qr) return seg[ind][id];
int mid = l + r >> 1;
return min(get(ind , ql , qr , lc , l , mid) , get(ind , ql , qr , rc , mid , r));
}
ll min_total_length(vector<int> r, vector<int> b) {
vector<pii> compress = {{-1 , 0} , {MOD , 0}};
int n = SZ(r) + SZ(b);
for(int i : r) compress.push_back({i , 0});
for(int i : b) compress.push_back({i , 1});
sort(all(compress));
dp[n] = 0;
int x = n , y = -1;
ll v = 0 , u = 0;
// for(int i = 0 ; i < SZ(compress) ; i++){
// cout << i << sep << compress[i].X << sep << compress[i].Y << endl;
// }
for(int i = n - 1 ; i >= 0 ; i--){
if(y == -1) dp[i] = INF;
else{
v += compress[x + 1].X - compress[i + 1].X;
u += compress[x].X - compress[i + 1].X;
dp[i] = min(get(0 , x + 1 , min(y + 1 , 2 * x - i + 1)) + v , get(1 , 2 * x - i + 1 , y + 1) + u);
}
// cout << i << sep << dp[i] << endl;
if(i > 0 && compress[i].Y != compress[i + 1].Y){
y = x; x = i; v = 0 , u = 0;
ll A = 0 , B = 0;
for(int j = i + 1 ; j <= y ; j++){
A += compress[j].X - compress[x + 1].X;
B += compress[j].X - compress[x].X;
modify(0 , j , dp[j] + A);
modify(1 , j , dp[j] + B);
dp[i] = min(dp[i] , dp[j] + B);
}
continue;
}
}
return dp[0];
}
Compilation message
wiring.cpp: In function 'void modify(int, int, ll, int, int, int)':
wiring.cpp:27:14: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
27 | int mid = l + r >> 1;
| ~~^~~
wiring.cpp: In function 'll get(int, int, int, int, int, int)':
wiring.cpp:36:14: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
36 | int mid = l + r >> 1;
| ~~^~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
424 KB |
Output is correct |
2 |
Correct |
1 ms |
332 KB |
Output is correct |
3 |
Correct |
1 ms |
308 KB |
Output is correct |
4 |
Correct |
1 ms |
332 KB |
Output is correct |
5 |
Correct |
1 ms |
332 KB |
Output is correct |
6 |
Correct |
1 ms |
332 KB |
Output is correct |
7 |
Correct |
1 ms |
332 KB |
Output is correct |
8 |
Correct |
1 ms |
332 KB |
Output is correct |
9 |
Correct |
1 ms |
332 KB |
Output is correct |
10 |
Correct |
1 ms |
332 KB |
Output is correct |
11 |
Correct |
1 ms |
332 KB |
Output is correct |
12 |
Correct |
1 ms |
332 KB |
Output is correct |
13 |
Correct |
2 ms |
332 KB |
Output is correct |
14 |
Correct |
1 ms |
332 KB |
Output is correct |
15 |
Correct |
1 ms |
332 KB |
Output is correct |
16 |
Correct |
1 ms |
436 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
332 KB |
Output is correct |
2 |
Correct |
1 ms |
332 KB |
Output is correct |
3 |
Correct |
107 ms |
6092 KB |
Output is correct |
4 |
Correct |
106 ms |
6124 KB |
Output is correct |
5 |
Correct |
100 ms |
8044 KB |
Output is correct |
6 |
Correct |
144 ms |
9260 KB |
Output is correct |
7 |
Correct |
140 ms |
9232 KB |
Output is correct |
8 |
Correct |
142 ms |
9268 KB |
Output is correct |
9 |
Correct |
140 ms |
9244 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
332 KB |
Output is correct |
2 |
Correct |
1 ms |
336 KB |
Output is correct |
3 |
Correct |
212 ms |
13236 KB |
Output is correct |
4 |
Correct |
227 ms |
13540 KB |
Output is correct |
5 |
Correct |
1 ms |
332 KB |
Output is correct |
6 |
Correct |
1 ms |
332 KB |
Output is correct |
7 |
Correct |
1 ms |
312 KB |
Output is correct |
8 |
Correct |
1 ms |
332 KB |
Output is correct |
9 |
Correct |
1 ms |
332 KB |
Output is correct |
10 |
Correct |
2 ms |
332 KB |
Output is correct |
11 |
Correct |
1 ms |
332 KB |
Output is correct |
12 |
Correct |
2 ms |
332 KB |
Output is correct |
13 |
Correct |
1 ms |
332 KB |
Output is correct |
14 |
Correct |
1 ms |
332 KB |
Output is correct |
15 |
Correct |
1 ms |
332 KB |
Output is correct |
16 |
Correct |
1 ms |
436 KB |
Output is correct |
17 |
Correct |
1 ms |
332 KB |
Output is correct |
18 |
Correct |
214 ms |
13476 KB |
Output is correct |
19 |
Correct |
209 ms |
13492 KB |
Output is correct |
20 |
Correct |
239 ms |
13480 KB |
Output is correct |
21 |
Correct |
210 ms |
13504 KB |
Output is correct |
22 |
Correct |
210 ms |
13492 KB |
Output is correct |
23 |
Correct |
219 ms |
13428 KB |
Output is correct |
24 |
Correct |
213 ms |
13432 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
332 KB |
Output is correct |
2 |
Correct |
216 ms |
13104 KB |
Output is correct |
3 |
Correct |
206 ms |
13340 KB |
Output is correct |
4 |
Correct |
215 ms |
13232 KB |
Output is correct |
5 |
Correct |
207 ms |
13296 KB |
Output is correct |
6 |
Correct |
1 ms |
332 KB |
Output is correct |
7 |
Correct |
1 ms |
332 KB |
Output is correct |
8 |
Correct |
1 ms |
332 KB |
Output is correct |
9 |
Correct |
1 ms |
332 KB |
Output is correct |
10 |
Correct |
1 ms |
332 KB |
Output is correct |
11 |
Correct |
1 ms |
332 KB |
Output is correct |
12 |
Correct |
1 ms |
304 KB |
Output is correct |
13 |
Correct |
2 ms |
332 KB |
Output is correct |
14 |
Correct |
1 ms |
332 KB |
Output is correct |
15 |
Correct |
1 ms |
432 KB |
Output is correct |
16 |
Correct |
1 ms |
332 KB |
Output is correct |
17 |
Correct |
1 ms |
332 KB |
Output is correct |
18 |
Correct |
220 ms |
12956 KB |
Output is correct |
19 |
Correct |
222 ms |
14492 KB |
Output is correct |
20 |
Correct |
226 ms |
14516 KB |
Output is correct |
21 |
Correct |
246 ms |
14020 KB |
Output is correct |
22 |
Correct |
196 ms |
12212 KB |
Output is correct |
23 |
Correct |
181 ms |
11020 KB |
Output is correct |
24 |
Correct |
201 ms |
12060 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
424 KB |
Output is correct |
2 |
Correct |
1 ms |
332 KB |
Output is correct |
3 |
Correct |
1 ms |
308 KB |
Output is correct |
4 |
Correct |
1 ms |
332 KB |
Output is correct |
5 |
Correct |
1 ms |
332 KB |
Output is correct |
6 |
Correct |
1 ms |
332 KB |
Output is correct |
7 |
Correct |
1 ms |
332 KB |
Output is correct |
8 |
Correct |
1 ms |
332 KB |
Output is correct |
9 |
Correct |
1 ms |
332 KB |
Output is correct |
10 |
Correct |
1 ms |
332 KB |
Output is correct |
11 |
Correct |
1 ms |
332 KB |
Output is correct |
12 |
Correct |
1 ms |
332 KB |
Output is correct |
13 |
Correct |
2 ms |
332 KB |
Output is correct |
14 |
Correct |
1 ms |
332 KB |
Output is correct |
15 |
Correct |
1 ms |
332 KB |
Output is correct |
16 |
Correct |
1 ms |
436 KB |
Output is correct |
17 |
Correct |
3 ms |
332 KB |
Output is correct |
18 |
Correct |
1 ms |
332 KB |
Output is correct |
19 |
Correct |
107 ms |
6092 KB |
Output is correct |
20 |
Correct |
106 ms |
6124 KB |
Output is correct |
21 |
Correct |
100 ms |
8044 KB |
Output is correct |
22 |
Correct |
144 ms |
9260 KB |
Output is correct |
23 |
Correct |
140 ms |
9232 KB |
Output is correct |
24 |
Correct |
142 ms |
9268 KB |
Output is correct |
25 |
Correct |
140 ms |
9244 KB |
Output is correct |
26 |
Correct |
1 ms |
332 KB |
Output is correct |
27 |
Correct |
1 ms |
336 KB |
Output is correct |
28 |
Correct |
212 ms |
13236 KB |
Output is correct |
29 |
Correct |
227 ms |
13540 KB |
Output is correct |
30 |
Correct |
1 ms |
332 KB |
Output is correct |
31 |
Correct |
1 ms |
332 KB |
Output is correct |
32 |
Correct |
1 ms |
312 KB |
Output is correct |
33 |
Correct |
1 ms |
332 KB |
Output is correct |
34 |
Correct |
1 ms |
332 KB |
Output is correct |
35 |
Correct |
2 ms |
332 KB |
Output is correct |
36 |
Correct |
1 ms |
332 KB |
Output is correct |
37 |
Correct |
2 ms |
332 KB |
Output is correct |
38 |
Correct |
1 ms |
332 KB |
Output is correct |
39 |
Correct |
1 ms |
332 KB |
Output is correct |
40 |
Correct |
1 ms |
332 KB |
Output is correct |
41 |
Correct |
1 ms |
436 KB |
Output is correct |
42 |
Correct |
1 ms |
332 KB |
Output is correct |
43 |
Correct |
214 ms |
13476 KB |
Output is correct |
44 |
Correct |
209 ms |
13492 KB |
Output is correct |
45 |
Correct |
239 ms |
13480 KB |
Output is correct |
46 |
Correct |
210 ms |
13504 KB |
Output is correct |
47 |
Correct |
210 ms |
13492 KB |
Output is correct |
48 |
Correct |
219 ms |
13428 KB |
Output is correct |
49 |
Correct |
213 ms |
13432 KB |
Output is correct |
50 |
Correct |
1 ms |
332 KB |
Output is correct |
51 |
Correct |
216 ms |
13104 KB |
Output is correct |
52 |
Correct |
206 ms |
13340 KB |
Output is correct |
53 |
Correct |
215 ms |
13232 KB |
Output is correct |
54 |
Correct |
207 ms |
13296 KB |
Output is correct |
55 |
Correct |
1 ms |
332 KB |
Output is correct |
56 |
Correct |
1 ms |
332 KB |
Output is correct |
57 |
Correct |
1 ms |
332 KB |
Output is correct |
58 |
Correct |
1 ms |
332 KB |
Output is correct |
59 |
Correct |
1 ms |
332 KB |
Output is correct |
60 |
Correct |
1 ms |
332 KB |
Output is correct |
61 |
Correct |
1 ms |
304 KB |
Output is correct |
62 |
Correct |
2 ms |
332 KB |
Output is correct |
63 |
Correct |
1 ms |
332 KB |
Output is correct |
64 |
Correct |
1 ms |
432 KB |
Output is correct |
65 |
Correct |
1 ms |
332 KB |
Output is correct |
66 |
Correct |
1 ms |
332 KB |
Output is correct |
67 |
Correct |
220 ms |
12956 KB |
Output is correct |
68 |
Correct |
222 ms |
14492 KB |
Output is correct |
69 |
Correct |
226 ms |
14516 KB |
Output is correct |
70 |
Correct |
246 ms |
14020 KB |
Output is correct |
71 |
Correct |
196 ms |
12212 KB |
Output is correct |
72 |
Correct |
181 ms |
11020 KB |
Output is correct |
73 |
Correct |
201 ms |
12060 KB |
Output is correct |
74 |
Correct |
182 ms |
11172 KB |
Output is correct |
75 |
Correct |
188 ms |
12904 KB |
Output is correct |
76 |
Correct |
184 ms |
11872 KB |
Output is correct |
77 |
Correct |
265 ms |
13796 KB |
Output is correct |
78 |
Correct |
246 ms |
12436 KB |
Output is correct |
79 |
Correct |
210 ms |
14200 KB |
Output is correct |
80 |
Correct |
187 ms |
11808 KB |
Output is correct |
81 |
Correct |
143 ms |
10984 KB |
Output is correct |
82 |
Correct |
171 ms |
10904 KB |
Output is correct |
83 |
Correct |
163 ms |
11296 KB |
Output is correct |
84 |
Correct |
199 ms |
10788 KB |
Output is correct |
85 |
Correct |
207 ms |
13380 KB |
Output is correct |
86 |
Correct |
187 ms |
11956 KB |
Output is correct |
87 |
Correct |
195 ms |
11736 KB |
Output is correct |
88 |
Correct |
212 ms |
12316 KB |
Output is correct |
89 |
Correct |
204 ms |
12388 KB |
Output is correct |
90 |
Correct |
223 ms |
12972 KB |
Output is correct |
91 |
Correct |
189 ms |
11136 KB |
Output is correct |
92 |
Correct |
206 ms |
13200 KB |
Output is correct |
93 |
Correct |
267 ms |
11808 KB |
Output is correct |
94 |
Correct |
215 ms |
14084 KB |
Output is correct |
95 |
Correct |
225 ms |
13860 KB |
Output is correct |
96 |
Correct |
179 ms |
11444 KB |
Output is correct |
97 |
Correct |
210 ms |
11232 KB |
Output is correct |
98 |
Correct |
200 ms |
12108 KB |
Output is correct |
99 |
Correct |
200 ms |
12040 KB |
Output is correct |
100 |
Correct |
231 ms |
14316 KB |
Output is correct |
101 |
Correct |
185 ms |
11296 KB |
Output is correct |
102 |
Correct |
181 ms |
11700 KB |
Output is correct |