// Source: https://oj.uz/problem/view/IOI17_books
//
#include "books.h"
#include "bits/stdc++.h"
using namespace std;
#define s second
#define f first
#define pb push_back
typedef long long ll;
typedef pair<int, int> pii;
typedef vector<pii> vpii;
typedef vector<int> vi;
#define FOR(i, a, b) for (int i = (a); i<b; i++)
bool ckmin(int& a, int b){ return b < a ? a = b, true : false; }
bool ckmax(int& a, int b){ return b > a ? a = b, true : false; }
const int N = 1e6 + 10;
int ri[N], le[N];
bool vis[N];
ll minimum_walk(vector<int> p, int s) {
// if (s == 0) {
ll ans = 0;
int n = p.size();
FOR(i, 0, n) {
if (!vis[i]) {
int l, r;
r = l = i;
int cur = p[i];
while (cur != i) {
l = min(l, cur);r=max(r, cur);
cur=p[cur];
}
cur = i;
do {
ans += abs(p[cur]-cur);
le[cur] = l;
ri[cur] = r;
vis[cur]=true;
cur=p[cur];
} while (cur != i);
}
}
int en = n-1;
while (en != -1 && p[en] == en) en--;
int st = 0;
while (st != n && p[st] == st) st++;
if (st > en) return 0;
// cout << ans <<
ans += 2 * (max(st - s, 0) + max(s - en, 0));
s = min(max(s, st), en);
int l = s, r = s;
// cout << ans << endl;
while (l > st || r < en) {
// cout << l << r << endl;
int tr = r;
int hr = ri[tr];
int tl = le[tr];
int cr = 0;
while (tr < en && (tl >= s || tr != hr)) {
// cout << tl << tr << endl;
if (tr == hr) cr+=2;
tr++;
tl=min(tl,le[tr]);
hr = max(hr, ri[tr]);
}
int right = tr;
// if (tl >= s) ans += cr; // didn't get through
tl = l;
int hl = le[tr];
tr = ri[tr];
int cl = 0;
while (tl > st && (tr <= s || tl != hl)) {
// cout << tl << l << endl;
if (tl == hl) cl+=2;
tl--;
tr=max(tr, ri[tl]);
hl = min(hl, le[tr]);
}
int left = tl;
// if (tr <= s) ans += cl; // didn't get through
// cout << cl << cr << endl;
// cout << ans << endl;
if (tl >= s && tr <= s) ans += cr + cl;
else if (tl >= s) ans += cr;
else if (tr <= s) ans += cl;
else ans += min(cl, cr);
// cout << ans << endl;
l = left, r = right;
}
// cout << ans << endl;
// int r =0;
// FOR(i, 0, en+1) {
// if (r < i) {
// r++;
// ans+=2;
// }
// r = max(p[i], r);
// }
return ans;
// } else assert(false);
}
// int main() {
// ios::sync_with_stdio(false);
// cin.tie(nullptr);
// int res = minimum_walk({0, 2, 3, 1},0);
// cout << res << endl;
// }
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
4440 KB |
Output is correct |
2 |
Correct |
1 ms |
4444 KB |
Output is correct |
3 |
Correct |
1 ms |
4696 KB |
Output is correct |
4 |
Correct |
1 ms |
4444 KB |
Output is correct |
5 |
Correct |
1 ms |
4444 KB |
Output is correct |
6 |
Correct |
1 ms |
4444 KB |
Output is correct |
7 |
Correct |
1 ms |
4444 KB |
Output is correct |
8 |
Correct |
1 ms |
4444 KB |
Output is correct |
9 |
Correct |
1 ms |
4444 KB |
Output is correct |
10 |
Correct |
1 ms |
4444 KB |
Output is correct |
11 |
Correct |
1 ms |
4444 KB |
Output is correct |
12 |
Correct |
1 ms |
4444 KB |
Output is correct |
13 |
Correct |
1 ms |
4444 KB |
Output is correct |
14 |
Correct |
1 ms |
4444 KB |
Output is correct |
15 |
Correct |
2 ms |
4440 KB |
Output is correct |
16 |
Correct |
1 ms |
4444 KB |
Output is correct |
17 |
Correct |
1 ms |
4440 KB |
Output is correct |
18 |
Correct |
1 ms |
4444 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
4440 KB |
Output is correct |
2 |
Correct |
1 ms |
4444 KB |
Output is correct |
3 |
Correct |
1 ms |
4696 KB |
Output is correct |
4 |
Correct |
1 ms |
4444 KB |
Output is correct |
5 |
Correct |
1 ms |
4444 KB |
Output is correct |
6 |
Correct |
1 ms |
4444 KB |
Output is correct |
7 |
Correct |
1 ms |
4444 KB |
Output is correct |
8 |
Correct |
1 ms |
4444 KB |
Output is correct |
9 |
Correct |
1 ms |
4444 KB |
Output is correct |
10 |
Correct |
1 ms |
4444 KB |
Output is correct |
11 |
Correct |
1 ms |
4444 KB |
Output is correct |
12 |
Correct |
1 ms |
4444 KB |
Output is correct |
13 |
Correct |
1 ms |
4444 KB |
Output is correct |
14 |
Correct |
1 ms |
4444 KB |
Output is correct |
15 |
Correct |
2 ms |
4440 KB |
Output is correct |
16 |
Correct |
1 ms |
4444 KB |
Output is correct |
17 |
Correct |
1 ms |
4440 KB |
Output is correct |
18 |
Correct |
1 ms |
4444 KB |
Output is correct |
19 |
Correct |
1 ms |
4444 KB |
Output is correct |
20 |
Correct |
1 ms |
4444 KB |
Output is correct |
21 |
Correct |
1 ms |
4444 KB |
Output is correct |
22 |
Correct |
1 ms |
4444 KB |
Output is correct |
23 |
Correct |
1 ms |
4536 KB |
Output is correct |
24 |
Correct |
1 ms |
4440 KB |
Output is correct |
25 |
Correct |
1 ms |
4444 KB |
Output is correct |
26 |
Correct |
1 ms |
4444 KB |
Output is correct |
27 |
Correct |
1 ms |
4444 KB |
Output is correct |
28 |
Correct |
1 ms |
4444 KB |
Output is correct |
29 |
Correct |
1 ms |
4444 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
4440 KB |
Output is correct |
2 |
Correct |
1 ms |
4444 KB |
Output is correct |
3 |
Correct |
1 ms |
4696 KB |
Output is correct |
4 |
Correct |
1 ms |
4444 KB |
Output is correct |
5 |
Correct |
1 ms |
4444 KB |
Output is correct |
6 |
Correct |
1 ms |
4444 KB |
Output is correct |
7 |
Correct |
1 ms |
4444 KB |
Output is correct |
8 |
Correct |
1 ms |
4444 KB |
Output is correct |
9 |
Correct |
1 ms |
4444 KB |
Output is correct |
10 |
Correct |
1 ms |
4444 KB |
Output is correct |
11 |
Correct |
1 ms |
4444 KB |
Output is correct |
12 |
Correct |
1 ms |
4444 KB |
Output is correct |
13 |
Correct |
1 ms |
4444 KB |
Output is correct |
14 |
Correct |
1 ms |
4444 KB |
Output is correct |
15 |
Correct |
2 ms |
4440 KB |
Output is correct |
16 |
Correct |
1 ms |
4444 KB |
Output is correct |
17 |
Correct |
1 ms |
4440 KB |
Output is correct |
18 |
Correct |
1 ms |
4444 KB |
Output is correct |
19 |
Correct |
1 ms |
4444 KB |
Output is correct |
20 |
Correct |
1 ms |
4444 KB |
Output is correct |
21 |
Correct |
1 ms |
4444 KB |
Output is correct |
22 |
Correct |
1 ms |
4444 KB |
Output is correct |
23 |
Correct |
1 ms |
4536 KB |
Output is correct |
24 |
Correct |
1 ms |
4440 KB |
Output is correct |
25 |
Correct |
1 ms |
4444 KB |
Output is correct |
26 |
Correct |
1 ms |
4444 KB |
Output is correct |
27 |
Correct |
1 ms |
4444 KB |
Output is correct |
28 |
Correct |
1 ms |
4444 KB |
Output is correct |
29 |
Correct |
1 ms |
4444 KB |
Output is correct |
30 |
Correct |
112 ms |
20468 KB |
Output is correct |
31 |
Correct |
115 ms |
20736 KB |
Output is correct |
32 |
Correct |
73 ms |
20472 KB |
Output is correct |
33 |
Correct |
77 ms |
20472 KB |
Output is correct |
34 |
Correct |
78 ms |
20560 KB |
Output is correct |
35 |
Correct |
78 ms |
20416 KB |
Output is correct |
36 |
Correct |
78 ms |
20484 KB |
Output is correct |
37 |
Correct |
76 ms |
20476 KB |
Output is correct |
38 |
Correct |
78 ms |
20560 KB |
Output is correct |
39 |
Correct |
77 ms |
20560 KB |
Output is correct |
40 |
Correct |
83 ms |
20820 KB |
Output is correct |
41 |
Correct |
95 ms |
20488 KB |
Output is correct |
42 |
Correct |
92 ms |
20492 KB |
Output is correct |
43 |
Correct |
78 ms |
20560 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
4444 KB |
3rd lines differ - on the 1st token, expected: '3304', found: '2744' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
4440 KB |
Output is correct |
2 |
Correct |
1 ms |
4444 KB |
Output is correct |
3 |
Correct |
1 ms |
4696 KB |
Output is correct |
4 |
Correct |
1 ms |
4444 KB |
Output is correct |
5 |
Correct |
1 ms |
4444 KB |
Output is correct |
6 |
Correct |
1 ms |
4444 KB |
Output is correct |
7 |
Correct |
1 ms |
4444 KB |
Output is correct |
8 |
Correct |
1 ms |
4444 KB |
Output is correct |
9 |
Correct |
1 ms |
4444 KB |
Output is correct |
10 |
Correct |
1 ms |
4444 KB |
Output is correct |
11 |
Correct |
1 ms |
4444 KB |
Output is correct |
12 |
Correct |
1 ms |
4444 KB |
Output is correct |
13 |
Correct |
1 ms |
4444 KB |
Output is correct |
14 |
Correct |
1 ms |
4444 KB |
Output is correct |
15 |
Correct |
2 ms |
4440 KB |
Output is correct |
16 |
Correct |
1 ms |
4444 KB |
Output is correct |
17 |
Correct |
1 ms |
4440 KB |
Output is correct |
18 |
Correct |
1 ms |
4444 KB |
Output is correct |
19 |
Correct |
1 ms |
4444 KB |
Output is correct |
20 |
Correct |
1 ms |
4444 KB |
Output is correct |
21 |
Correct |
1 ms |
4444 KB |
Output is correct |
22 |
Correct |
1 ms |
4444 KB |
Output is correct |
23 |
Correct |
1 ms |
4536 KB |
Output is correct |
24 |
Correct |
1 ms |
4440 KB |
Output is correct |
25 |
Correct |
1 ms |
4444 KB |
Output is correct |
26 |
Correct |
1 ms |
4444 KB |
Output is correct |
27 |
Correct |
1 ms |
4444 KB |
Output is correct |
28 |
Correct |
1 ms |
4444 KB |
Output is correct |
29 |
Correct |
1 ms |
4444 KB |
Output is correct |
30 |
Correct |
112 ms |
20468 KB |
Output is correct |
31 |
Correct |
115 ms |
20736 KB |
Output is correct |
32 |
Correct |
73 ms |
20472 KB |
Output is correct |
33 |
Correct |
77 ms |
20472 KB |
Output is correct |
34 |
Correct |
78 ms |
20560 KB |
Output is correct |
35 |
Correct |
78 ms |
20416 KB |
Output is correct |
36 |
Correct |
78 ms |
20484 KB |
Output is correct |
37 |
Correct |
76 ms |
20476 KB |
Output is correct |
38 |
Correct |
78 ms |
20560 KB |
Output is correct |
39 |
Correct |
77 ms |
20560 KB |
Output is correct |
40 |
Correct |
83 ms |
20820 KB |
Output is correct |
41 |
Correct |
95 ms |
20488 KB |
Output is correct |
42 |
Correct |
92 ms |
20492 KB |
Output is correct |
43 |
Correct |
78 ms |
20560 KB |
Output is correct |
44 |
Incorrect |
1 ms |
4444 KB |
3rd lines differ - on the 1st token, expected: '3304', found: '2744' |
45 |
Halted |
0 ms |
0 KB |
- |