#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define pii pair<int, int>
#define f first
#define s second
#define ALL(x) (x).begin(), (x).end()
#define SZ(x) (int)((x).size())
#define pb push_back
#ifdef BALBIT
#define bug(...) cerr<<"#"<<__LINE__<<" "<<#__VA_ARGS__<<": ", _do(__VA_ARGS__)
template <typename T> void _do(T && x) {cerr<<x<<endl;}
template <typename T, typename ...S> void _do(T && x, S&&...y) {cerr<<x<<", "; _do(y...);}
#define IOS()
#else
#define bug(...)
#define IOS() ios::sync_with_stdio(0), cin.tie(0)
#define endl '\n'
#endif
const int maxn = 1e6+5;
bool seen[maxn];
int R[maxn], L[maxn];
bool st[maxn];
bool dv[maxn];
int RR[maxn], LL[maxn];
int par[maxn];
int find(int x) {return x == par[x]? x : par[x] = find(par[x]);}
void mrg(int a, int b) {
a = find(a); b = find(b);
if (a == b) return;
RR[a] = max(RR[a], RR[b]);
LL[a] = min(LL[a], LL[b]);
par[b] = a;
}
ll minimum_walk(vector<int> p, int s) {
memset(seen, 0, sizeof seen);
int n = SZ(p);
ll re= 0;
for (int i = 0; i<n; ++i) {
// --p[i];
re+= abs(i-p[i]);
}
// vector<pair<pii,int> > v;
for (int i = 0; i<n; ++i) {
if (!seen[i]) {
st[i] = 1;
seen[i] = 1;
R[i] = L[i] = i;
for (int x = p[i]; x != i; x = p[x]) seen[x] = 1, R[i] = max(R[i], x);
for (int x = p[i]; x != i; x = p[x]) R[x] = R[i];
bug(i, L[i], R[i]);
LL[i] = L[i]; RR[i] =R[i];
}
}
int nmx = 0;
re -= 2;
for (int i = 0; i<n; ++i) {
nmx = max(nmx, R[i]);
if (nmx <= i) {
re += 2;
dv[i] = 1;
}
// if (i == s && nmx > i) {
// int ad = 0;
// for (int j = 1; ; ++j) {
// if (p[i+j] != i+j || p[i-j] != i-j) {
// ad = j; break;
// }
// }
// bug(ad);
// }
}
for (int i = 0; i<n; ++i) par[i] = i;
vector<pii> stk;
for (int i = 0; i<n; ++i) {
if (st[i]) {
while (!stk.empty() && i > stk.back().s) stk.pop_back();
int newl = i;
while (!stk.empty() && R[i] > stk.back().s) {
mrg(i, stk.back().f);
newl = stk.back().f;
stk.pop_back();
}
stk.pb({newl, R[i]});
}
}
{
int l= LL[find(s)], r = RR[find(s)];
for (int i=l-1; i>=0; --i) {
if (st[i] && RR[find(i)]>r) {
re += min(l-LL[find(i)], RR[find(i)] - r) * 2;
l = LL[find(i)]; r = RR[find(i)];
i = l-1;
}
}
}
for (int i = 0; i<n; ++i) {
if (p[i] == i && i != s) re -= 2;
else break;
}
for (int i = n-1; i>=0; --i) {
if (p[i] == i && i != s) re -= 2;
else break;
}
return re;
}
#ifdef BALBIT
signed main(){
IOS();
bug(minimum_walk({3,1,2,0},0));
// bug(minimum_walk({3,2,1,0},0));
// bug(minimum_walk({1,0,3,2},0));
}
#endif
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
1408 KB |
Output is correct |
2 |
Correct |
1 ms |
1280 KB |
Output is correct |
3 |
Correct |
2 ms |
1440 KB |
Output is correct |
4 |
Correct |
1 ms |
1280 KB |
Output is correct |
5 |
Correct |
1 ms |
1280 KB |
Output is correct |
6 |
Correct |
1 ms |
1280 KB |
Output is correct |
7 |
Correct |
1 ms |
1280 KB |
Output is correct |
8 |
Correct |
3 ms |
1280 KB |
Output is correct |
9 |
Correct |
2 ms |
1280 KB |
Output is correct |
10 |
Correct |
2 ms |
1280 KB |
Output is correct |
11 |
Correct |
2 ms |
1280 KB |
Output is correct |
12 |
Correct |
1 ms |
1280 KB |
Output is correct |
13 |
Correct |
1 ms |
1280 KB |
Output is correct |
14 |
Correct |
1 ms |
1280 KB |
Output is correct |
15 |
Correct |
1 ms |
1280 KB |
Output is correct |
16 |
Correct |
1 ms |
1280 KB |
Output is correct |
17 |
Correct |
1 ms |
1280 KB |
Output is correct |
18 |
Correct |
1 ms |
1280 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
1408 KB |
Output is correct |
2 |
Correct |
1 ms |
1280 KB |
Output is correct |
3 |
Correct |
2 ms |
1440 KB |
Output is correct |
4 |
Correct |
1 ms |
1280 KB |
Output is correct |
5 |
Correct |
1 ms |
1280 KB |
Output is correct |
6 |
Correct |
1 ms |
1280 KB |
Output is correct |
7 |
Correct |
1 ms |
1280 KB |
Output is correct |
8 |
Correct |
3 ms |
1280 KB |
Output is correct |
9 |
Correct |
2 ms |
1280 KB |
Output is correct |
10 |
Correct |
2 ms |
1280 KB |
Output is correct |
11 |
Correct |
2 ms |
1280 KB |
Output is correct |
12 |
Correct |
1 ms |
1280 KB |
Output is correct |
13 |
Correct |
1 ms |
1280 KB |
Output is correct |
14 |
Correct |
1 ms |
1280 KB |
Output is correct |
15 |
Correct |
1 ms |
1280 KB |
Output is correct |
16 |
Correct |
1 ms |
1280 KB |
Output is correct |
17 |
Correct |
1 ms |
1280 KB |
Output is correct |
18 |
Correct |
1 ms |
1280 KB |
Output is correct |
19 |
Correct |
1 ms |
1408 KB |
Output is correct |
20 |
Correct |
1 ms |
1408 KB |
Output is correct |
21 |
Correct |
1 ms |
1408 KB |
Output is correct |
22 |
Correct |
1 ms |
1408 KB |
Output is correct |
23 |
Correct |
1 ms |
1408 KB |
Output is correct |
24 |
Correct |
1 ms |
1408 KB |
Output is correct |
25 |
Correct |
1 ms |
1408 KB |
Output is correct |
26 |
Correct |
1 ms |
1408 KB |
Output is correct |
27 |
Correct |
1 ms |
1408 KB |
Output is correct |
28 |
Correct |
1 ms |
1408 KB |
Output is correct |
29 |
Correct |
1 ms |
1408 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
1408 KB |
Output is correct |
2 |
Correct |
1 ms |
1280 KB |
Output is correct |
3 |
Correct |
2 ms |
1440 KB |
Output is correct |
4 |
Correct |
1 ms |
1280 KB |
Output is correct |
5 |
Correct |
1 ms |
1280 KB |
Output is correct |
6 |
Correct |
1 ms |
1280 KB |
Output is correct |
7 |
Correct |
1 ms |
1280 KB |
Output is correct |
8 |
Correct |
3 ms |
1280 KB |
Output is correct |
9 |
Correct |
2 ms |
1280 KB |
Output is correct |
10 |
Correct |
2 ms |
1280 KB |
Output is correct |
11 |
Correct |
2 ms |
1280 KB |
Output is correct |
12 |
Correct |
1 ms |
1280 KB |
Output is correct |
13 |
Correct |
1 ms |
1280 KB |
Output is correct |
14 |
Correct |
1 ms |
1280 KB |
Output is correct |
15 |
Correct |
1 ms |
1280 KB |
Output is correct |
16 |
Correct |
1 ms |
1280 KB |
Output is correct |
17 |
Correct |
1 ms |
1280 KB |
Output is correct |
18 |
Correct |
1 ms |
1280 KB |
Output is correct |
19 |
Correct |
1 ms |
1408 KB |
Output is correct |
20 |
Correct |
1 ms |
1408 KB |
Output is correct |
21 |
Correct |
1 ms |
1408 KB |
Output is correct |
22 |
Correct |
1 ms |
1408 KB |
Output is correct |
23 |
Correct |
1 ms |
1408 KB |
Output is correct |
24 |
Correct |
1 ms |
1408 KB |
Output is correct |
25 |
Correct |
1 ms |
1408 KB |
Output is correct |
26 |
Correct |
1 ms |
1408 KB |
Output is correct |
27 |
Correct |
1 ms |
1408 KB |
Output is correct |
28 |
Correct |
1 ms |
1408 KB |
Output is correct |
29 |
Correct |
1 ms |
1408 KB |
Output is correct |
30 |
Correct |
311 ms |
20600 KB |
Output is correct |
31 |
Correct |
328 ms |
20532 KB |
Output is correct |
32 |
Correct |
173 ms |
34296 KB |
Output is correct |
33 |
Correct |
182 ms |
34168 KB |
Output is correct |
34 |
Correct |
181 ms |
34180 KB |
Output is correct |
35 |
Correct |
192 ms |
34296 KB |
Output is correct |
36 |
Correct |
188 ms |
34168 KB |
Output is correct |
37 |
Correct |
182 ms |
34296 KB |
Output is correct |
38 |
Correct |
180 ms |
32716 KB |
Output is correct |
39 |
Correct |
186 ms |
31864 KB |
Output is correct |
40 |
Correct |
207 ms |
24568 KB |
Output is correct |
41 |
Correct |
228 ms |
20940 KB |
Output is correct |
42 |
Correct |
207 ms |
21624 KB |
Output is correct |
43 |
Correct |
185 ms |
33144 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
1408 KB |
3rd lines differ - on the 1st token, expected: '3304', found: '3590' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
1408 KB |
Output is correct |
2 |
Correct |
1 ms |
1280 KB |
Output is correct |
3 |
Correct |
2 ms |
1440 KB |
Output is correct |
4 |
Correct |
1 ms |
1280 KB |
Output is correct |
5 |
Correct |
1 ms |
1280 KB |
Output is correct |
6 |
Correct |
1 ms |
1280 KB |
Output is correct |
7 |
Correct |
1 ms |
1280 KB |
Output is correct |
8 |
Correct |
3 ms |
1280 KB |
Output is correct |
9 |
Correct |
2 ms |
1280 KB |
Output is correct |
10 |
Correct |
2 ms |
1280 KB |
Output is correct |
11 |
Correct |
2 ms |
1280 KB |
Output is correct |
12 |
Correct |
1 ms |
1280 KB |
Output is correct |
13 |
Correct |
1 ms |
1280 KB |
Output is correct |
14 |
Correct |
1 ms |
1280 KB |
Output is correct |
15 |
Correct |
1 ms |
1280 KB |
Output is correct |
16 |
Correct |
1 ms |
1280 KB |
Output is correct |
17 |
Correct |
1 ms |
1280 KB |
Output is correct |
18 |
Correct |
1 ms |
1280 KB |
Output is correct |
19 |
Correct |
1 ms |
1408 KB |
Output is correct |
20 |
Correct |
1 ms |
1408 KB |
Output is correct |
21 |
Correct |
1 ms |
1408 KB |
Output is correct |
22 |
Correct |
1 ms |
1408 KB |
Output is correct |
23 |
Correct |
1 ms |
1408 KB |
Output is correct |
24 |
Correct |
1 ms |
1408 KB |
Output is correct |
25 |
Correct |
1 ms |
1408 KB |
Output is correct |
26 |
Correct |
1 ms |
1408 KB |
Output is correct |
27 |
Correct |
1 ms |
1408 KB |
Output is correct |
28 |
Correct |
1 ms |
1408 KB |
Output is correct |
29 |
Correct |
1 ms |
1408 KB |
Output is correct |
30 |
Correct |
311 ms |
20600 KB |
Output is correct |
31 |
Correct |
328 ms |
20532 KB |
Output is correct |
32 |
Correct |
173 ms |
34296 KB |
Output is correct |
33 |
Correct |
182 ms |
34168 KB |
Output is correct |
34 |
Correct |
181 ms |
34180 KB |
Output is correct |
35 |
Correct |
192 ms |
34296 KB |
Output is correct |
36 |
Correct |
188 ms |
34168 KB |
Output is correct |
37 |
Correct |
182 ms |
34296 KB |
Output is correct |
38 |
Correct |
180 ms |
32716 KB |
Output is correct |
39 |
Correct |
186 ms |
31864 KB |
Output is correct |
40 |
Correct |
207 ms |
24568 KB |
Output is correct |
41 |
Correct |
228 ms |
20940 KB |
Output is correct |
42 |
Correct |
207 ms |
21624 KB |
Output is correct |
43 |
Correct |
185 ms |
33144 KB |
Output is correct |
44 |
Incorrect |
1 ms |
1408 KB |
3rd lines differ - on the 1st token, expected: '3304', found: '3590' |
45 |
Halted |
0 ms |
0 KB |
- |