#include "books.h"
#include <bits/stdc++.h>
using namespace std;
long long minimum_walk(std::vector<int> p, int s) {
long long ans = 0;
int n = p.size();
for (int i = 0; i < n; i++) {
ans += abs(i - p[i]);
}
vector<bool> vis(n);
vector<int> mn(n, n), mx(n);
for (int i = 0; i < n; i++) {
if (vis[i]) continue;
int cur = i;
vector<int> all;
while (!vis[cur]) {
all.push_back(cur);
vis[cur] = true;
mn[i] = min(mn[i], cur);
mx[i] = max(mx[i], cur);
cur = p[cur];
}
for (int j : all) mn[j] = mn[i], mx[j] = mx[i];
}
int L = mn[s], R = mx[s], l = s, r = s;
auto simulate = [&]() {
while (L < l || r < R) {
if (L < l) {
L = min(L, mn[l]);
R = max(R, mx[l]);
l--;
} else {
L = min(L, mn[r]);
R = max(R, mx[r]);
r++;
}
}
};
int lto = 0, rto = n - 1;
for (int i = 0; i < n; i++) {
if (p[i] == i) lto++;
else break;
}
for (int i = n - 1; i >= 0; i--) {
if (p[i] == i) rto--;
else break;
}
lto = min(lto, s);
rto = max(rto, s);
simulate();
while (true) {
bool first = false;
int right_cost = 0;
int m = r;
for (int i = r + 1; i <= rto; i++) {
if (mn[i] < l) {
if (i > m) right_cost++;
first = true;
break;
}
if (mn[i] > m) right_cost++;
m = max(m, mx[i]);
}
bool second = false;
int left_cost = 0;
m = l;
for (int i = l - 1; i >= lto; i--) {
if (mx[i] > r) {
if (i < m) left_cost++;
second = true;
break;
}
if (mx[i] < m) left_cost++;
m = min(m, mn[i]);
}
if (!first) {
ans += 2 * (left_cost + right_cost);
break;
}
if (left_cost < right_cost) {
ans += 2 * left_cost;
for (int i = r + 1; i <= rto; i++) {
L = min(L, mn[i]), R = max(R, mx[i]);
if (mn[i] < l) break;
}
simulate();
} else {
ans += 2 * right_cost;
for (int i = l - 1; i >= lto; i--) { L = min(L, mn[i]), R = max(R, mx[i]);
if (mx[i] > r) break;
}
simulate();
}
}
return ans;
}
Compilation message
books.cpp: In function 'long long int minimum_walk(std::vector<int>, int)':
books.cpp:64:8: warning: variable 'second' set but not used [-Wunused-but-set-variable]
64 | bool second = false;
| ^~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
344 KB |
Output is correct |
2 |
Correct |
0 ms |
344 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
0 ms |
348 KB |
Output is correct |
5 |
Correct |
0 ms |
348 KB |
Output is correct |
6 |
Correct |
0 ms |
348 KB |
Output is correct |
7 |
Correct |
0 ms |
348 KB |
Output is correct |
8 |
Correct |
0 ms |
348 KB |
Output is correct |
9 |
Correct |
0 ms |
348 KB |
Output is correct |
10 |
Correct |
0 ms |
348 KB |
Output is correct |
11 |
Correct |
0 ms |
348 KB |
Output is correct |
12 |
Correct |
0 ms |
348 KB |
Output is correct |
13 |
Correct |
1 ms |
344 KB |
Output is correct |
14 |
Correct |
0 ms |
348 KB |
Output is correct |
15 |
Correct |
0 ms |
348 KB |
Output is correct |
16 |
Correct |
0 ms |
348 KB |
Output is correct |
17 |
Correct |
0 ms |
348 KB |
Output is correct |
18 |
Correct |
0 ms |
348 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
344 KB |
Output is correct |
2 |
Correct |
0 ms |
344 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
0 ms |
348 KB |
Output is correct |
5 |
Correct |
0 ms |
348 KB |
Output is correct |
6 |
Correct |
0 ms |
348 KB |
Output is correct |
7 |
Correct |
0 ms |
348 KB |
Output is correct |
8 |
Correct |
0 ms |
348 KB |
Output is correct |
9 |
Correct |
0 ms |
348 KB |
Output is correct |
10 |
Correct |
0 ms |
348 KB |
Output is correct |
11 |
Correct |
0 ms |
348 KB |
Output is correct |
12 |
Correct |
0 ms |
348 KB |
Output is correct |
13 |
Correct |
1 ms |
344 KB |
Output is correct |
14 |
Correct |
0 ms |
348 KB |
Output is correct |
15 |
Correct |
0 ms |
348 KB |
Output is correct |
16 |
Correct |
0 ms |
348 KB |
Output is correct |
17 |
Correct |
0 ms |
348 KB |
Output is correct |
18 |
Correct |
0 ms |
348 KB |
Output is correct |
19 |
Correct |
0 ms |
348 KB |
Output is correct |
20 |
Correct |
0 ms |
348 KB |
Output is correct |
21 |
Correct |
0 ms |
348 KB |
Output is correct |
22 |
Correct |
0 ms |
348 KB |
Output is correct |
23 |
Correct |
0 ms |
348 KB |
Output is correct |
24 |
Correct |
1 ms |
348 KB |
Output is correct |
25 |
Correct |
0 ms |
348 KB |
Output is correct |
26 |
Correct |
0 ms |
348 KB |
Output is correct |
27 |
Correct |
0 ms |
348 KB |
Output is correct |
28 |
Correct |
0 ms |
348 KB |
Output is correct |
29 |
Correct |
0 ms |
348 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
344 KB |
Output is correct |
2 |
Correct |
0 ms |
344 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
0 ms |
348 KB |
Output is correct |
5 |
Correct |
0 ms |
348 KB |
Output is correct |
6 |
Correct |
0 ms |
348 KB |
Output is correct |
7 |
Correct |
0 ms |
348 KB |
Output is correct |
8 |
Correct |
0 ms |
348 KB |
Output is correct |
9 |
Correct |
0 ms |
348 KB |
Output is correct |
10 |
Correct |
0 ms |
348 KB |
Output is correct |
11 |
Correct |
0 ms |
348 KB |
Output is correct |
12 |
Correct |
0 ms |
348 KB |
Output is correct |
13 |
Correct |
1 ms |
344 KB |
Output is correct |
14 |
Correct |
0 ms |
348 KB |
Output is correct |
15 |
Correct |
0 ms |
348 KB |
Output is correct |
16 |
Correct |
0 ms |
348 KB |
Output is correct |
17 |
Correct |
0 ms |
348 KB |
Output is correct |
18 |
Correct |
0 ms |
348 KB |
Output is correct |
19 |
Correct |
0 ms |
348 KB |
Output is correct |
20 |
Correct |
0 ms |
348 KB |
Output is correct |
21 |
Correct |
0 ms |
348 KB |
Output is correct |
22 |
Correct |
0 ms |
348 KB |
Output is correct |
23 |
Correct |
0 ms |
348 KB |
Output is correct |
24 |
Correct |
1 ms |
348 KB |
Output is correct |
25 |
Correct |
0 ms |
348 KB |
Output is correct |
26 |
Correct |
0 ms |
348 KB |
Output is correct |
27 |
Correct |
0 ms |
348 KB |
Output is correct |
28 |
Correct |
0 ms |
348 KB |
Output is correct |
29 |
Correct |
0 ms |
348 KB |
Output is correct |
30 |
Correct |
94 ms |
20340 KB |
Output is correct |
31 |
Correct |
94 ms |
18292 KB |
Output is correct |
32 |
Correct |
84 ms |
15952 KB |
Output is correct |
33 |
Correct |
90 ms |
15956 KB |
Output is correct |
34 |
Correct |
91 ms |
15956 KB |
Output is correct |
35 |
Correct |
93 ms |
16128 KB |
Output is correct |
36 |
Correct |
93 ms |
16212 KB |
Output is correct |
37 |
Correct |
75 ms |
16208 KB |
Output is correct |
38 |
Correct |
75 ms |
16124 KB |
Output is correct |
39 |
Correct |
73 ms |
16208 KB |
Output is correct |
40 |
Correct |
78 ms |
16444 KB |
Output is correct |
41 |
Correct |
92 ms |
19212 KB |
Output is correct |
42 |
Correct |
89 ms |
17784 KB |
Output is correct |
43 |
Correct |
96 ms |
16208 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
348 KB |
Output is correct |
2 |
Correct |
0 ms |
348 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
0 ms |
348 KB |
Output is correct |
5 |
Correct |
0 ms |
348 KB |
Output is correct |
6 |
Correct |
0 ms |
348 KB |
Output is correct |
7 |
Correct |
0 ms |
348 KB |
Output is correct |
8 |
Correct |
0 ms |
348 KB |
Output is correct |
9 |
Correct |
0 ms |
348 KB |
Output is correct |
10 |
Correct |
0 ms |
348 KB |
Output is correct |
11 |
Correct |
1 ms |
348 KB |
Output is correct |
12 |
Correct |
0 ms |
348 KB |
Output is correct |
13 |
Correct |
0 ms |
348 KB |
Output is correct |
14 |
Correct |
0 ms |
348 KB |
Output is correct |
15 |
Correct |
0 ms |
348 KB |
Output is correct |
16 |
Correct |
0 ms |
348 KB |
Output is correct |
17 |
Correct |
0 ms |
348 KB |
Output is correct |
18 |
Correct |
0 ms |
348 KB |
Output is correct |
19 |
Correct |
1 ms |
348 KB |
Output is correct |
20 |
Correct |
0 ms |
348 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
344 KB |
Output is correct |
2 |
Correct |
0 ms |
344 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
0 ms |
348 KB |
Output is correct |
5 |
Correct |
0 ms |
348 KB |
Output is correct |
6 |
Correct |
0 ms |
348 KB |
Output is correct |
7 |
Correct |
0 ms |
348 KB |
Output is correct |
8 |
Correct |
0 ms |
348 KB |
Output is correct |
9 |
Correct |
0 ms |
348 KB |
Output is correct |
10 |
Correct |
0 ms |
348 KB |
Output is correct |
11 |
Correct |
0 ms |
348 KB |
Output is correct |
12 |
Correct |
0 ms |
348 KB |
Output is correct |
13 |
Correct |
1 ms |
344 KB |
Output is correct |
14 |
Correct |
0 ms |
348 KB |
Output is correct |
15 |
Correct |
0 ms |
348 KB |
Output is correct |
16 |
Correct |
0 ms |
348 KB |
Output is correct |
17 |
Correct |
0 ms |
348 KB |
Output is correct |
18 |
Correct |
0 ms |
348 KB |
Output is correct |
19 |
Correct |
0 ms |
348 KB |
Output is correct |
20 |
Correct |
0 ms |
348 KB |
Output is correct |
21 |
Correct |
0 ms |
348 KB |
Output is correct |
22 |
Correct |
0 ms |
348 KB |
Output is correct |
23 |
Correct |
0 ms |
348 KB |
Output is correct |
24 |
Correct |
1 ms |
348 KB |
Output is correct |
25 |
Correct |
0 ms |
348 KB |
Output is correct |
26 |
Correct |
0 ms |
348 KB |
Output is correct |
27 |
Correct |
0 ms |
348 KB |
Output is correct |
28 |
Correct |
0 ms |
348 KB |
Output is correct |
29 |
Correct |
0 ms |
348 KB |
Output is correct |
30 |
Correct |
94 ms |
20340 KB |
Output is correct |
31 |
Correct |
94 ms |
18292 KB |
Output is correct |
32 |
Correct |
84 ms |
15952 KB |
Output is correct |
33 |
Correct |
90 ms |
15956 KB |
Output is correct |
34 |
Correct |
91 ms |
15956 KB |
Output is correct |
35 |
Correct |
93 ms |
16128 KB |
Output is correct |
36 |
Correct |
93 ms |
16212 KB |
Output is correct |
37 |
Correct |
75 ms |
16208 KB |
Output is correct |
38 |
Correct |
75 ms |
16124 KB |
Output is correct |
39 |
Correct |
73 ms |
16208 KB |
Output is correct |
40 |
Correct |
78 ms |
16444 KB |
Output is correct |
41 |
Correct |
92 ms |
19212 KB |
Output is correct |
42 |
Correct |
89 ms |
17784 KB |
Output is correct |
43 |
Correct |
96 ms |
16208 KB |
Output is correct |
44 |
Correct |
1 ms |
348 KB |
Output is correct |
45 |
Correct |
0 ms |
348 KB |
Output is correct |
46 |
Correct |
0 ms |
348 KB |
Output is correct |
47 |
Correct |
0 ms |
348 KB |
Output is correct |
48 |
Correct |
0 ms |
348 KB |
Output is correct |
49 |
Correct |
0 ms |
348 KB |
Output is correct |
50 |
Correct |
0 ms |
348 KB |
Output is correct |
51 |
Correct |
0 ms |
348 KB |
Output is correct |
52 |
Correct |
0 ms |
348 KB |
Output is correct |
53 |
Correct |
0 ms |
348 KB |
Output is correct |
54 |
Correct |
1 ms |
348 KB |
Output is correct |
55 |
Correct |
0 ms |
348 KB |
Output is correct |
56 |
Correct |
0 ms |
348 KB |
Output is correct |
57 |
Correct |
0 ms |
348 KB |
Output is correct |
58 |
Correct |
0 ms |
348 KB |
Output is correct |
59 |
Correct |
0 ms |
348 KB |
Output is correct |
60 |
Correct |
0 ms |
348 KB |
Output is correct |
61 |
Correct |
0 ms |
348 KB |
Output is correct |
62 |
Correct |
1 ms |
348 KB |
Output is correct |
63 |
Correct |
0 ms |
348 KB |
Output is correct |
64 |
Correct |
89 ms |
15956 KB |
Output is correct |
65 |
Correct |
97 ms |
16112 KB |
Output is correct |
66 |
Correct |
78 ms |
16108 KB |
Output is correct |
67 |
Correct |
72 ms |
16128 KB |
Output is correct |
68 |
Correct |
8 ms |
1884 KB |
Output is correct |
69 |
Correct |
8 ms |
1884 KB |
Output is correct |
70 |
Correct |
7 ms |
1884 KB |
Output is correct |
71 |
Correct |
9 ms |
1856 KB |
Output is correct |
72 |
Correct |
9 ms |
1884 KB |
Output is correct |
73 |
Correct |
7 ms |
1884 KB |
Output is correct |
74 |
Correct |
83 ms |
16104 KB |
Output is correct |
75 |
Correct |
90 ms |
16208 KB |
Output is correct |
76 |
Correct |
84 ms |
16212 KB |
Output is correct |
77 |
Correct |
92 ms |
16216 KB |
Output is correct |
78 |
Correct |
91 ms |
16128 KB |
Output is correct |
79 |
Correct |
101 ms |
15956 KB |
Output is correct |
80 |
Correct |
80 ms |
16112 KB |
Output is correct |
81 |
Correct |
98 ms |
15952 KB |
Output is correct |
82 |
Correct |
93 ms |
22864 KB |
Output is correct |
83 |
Correct |
100 ms |
22888 KB |
Output is correct |
84 |
Correct |
99 ms |
22772 KB |
Output is correct |
85 |
Correct |
87 ms |
22864 KB |
Output is correct |
86 |
Correct |
82 ms |
22884 KB |
Output is correct |
87 |
Correct |
97 ms |
22896 KB |
Output is correct |
88 |
Correct |
95 ms |
22868 KB |
Output is correct |
89 |
Correct |
92 ms |
22764 KB |
Output is correct |
90 |
Correct |
79 ms |
22904 KB |
Output is correct |
91 |
Correct |
83 ms |
22900 KB |
Output is correct |
92 |
Correct |
89 ms |
22864 KB |
Output is correct |