#include "books.h"
#include <bits/stdc++.h>
#define pb push_back
#define x first
#define y second
#define all(a) a.begin(), a.end()
#define sz(a) (int)a.size()
#define rd() abs((int)rng())
using namespace std;
typedef long long ll;
typedef long double ld;
typedef pair<int, int> pii;
const int maxn = 1e6 + 100;
const int mod = 1e9 + 7;
mt19937 rng(chrono::high_resolution_clock::now().time_since_epoch().count());
int n, s, val[maxn], nex, L[maxn], R[maxn], fr, to, prv[maxn], nxt[maxn], addprv[maxn], addnxt[maxn];
ll base;
vector<int>p;
pii extend(int l, int r)
{
int cur_l = l, cur_r = r;
int i = l, j = r;
while(i >= cur_l || j <= cur_r)
{
if(i >= cur_l)
{
cur_l = min(cur_l, L[val[i]]);
cur_r = max(cur_r, R[val[i]]);
--i;
}
if(j <= cur_r)
{
cur_l = min(cur_l, L[val[j]]);
cur_r = max(cur_r, R[val[j]]);
j++;
}
}
return {cur_l, cur_r};
}
ll solve(int l, int r)
{
pii aa = extend(l, r);
int cur_l = aa.x, cur_r = aa.y;
if(cur_l <= fr && cur_r >= to)
return 0;
if(prv[cur_l] == -1)
{
if(cur_l > fr)
return 2 + solve(cur_l - 1, cur_r);
else
return 2 + solve(cur_l, cur_r + 1);
}
if(addprv[cur_l] < addnxt[cur_r])
return addprv[cur_l] + solve(prv[cur_l], cur_r);
else
return addnxt[cur_r] + solve(cur_l, nxt[cur_r]);
}
ll minimum_walk(vector<int>perm, int S)
{
p = perm;
s = S;
n = sz(p);
for(int i = 1; i <= n; i++)
L[i] = mod, R[i] = -mod;
to = n - 1, fr = 0;
while(to >= 0 && p[to] == to)
--to;
while(fr < n && p[fr] == fr)
++fr;
if(fr > to)
return 0;
for(int i = 0; i < n; i++)
{
if(val[i] == 0)
{
int col = ++nex, cur = i;
while(val[cur] == 0)
{
val[cur] = col;
L[col] = min(L[col], cur);
R[col] = max(R[col], cur);
base += abs(p[cur] - cur);
cur = p[cur];
}
}
}
vector<int>cnt(n, 0);
for(int i = 1; i <= nex; i++)
cnt[L[i]]++, cnt[R[i]]--;
int pos = -1, bal = 0, cadd = 0;
for(int i = 0; i <= s; i++)
{
prv[i] = pos, addprv[i] = cadd;
if(R[val[i]] > s)
{
pos = i;
cadd = 0;
bal = 0;
}
else
bal += cnt[i];
if(bal == 0)
cadd += 2;
}
pos = -1, bal = 0, cadd = 0;
for(int i = n - 1; i >= s; i--)
{
nxt[i] = pos, addnxt[i] = cadd;
if(L[val[i]] < s)
{
pos = i;
cadd = 0;
bal = 0;
}
else
bal += cnt[i];
if(bal == 0)
cadd += 2;
}
return base + solve(s, s);
}
/*int main()
{
ios_base::sync_with_stdio(false), cin.tie(0);
vector<int>perm;
ifstream fin("in.txt");
int n, s;
fin >> n >> s;
perm.resize(n);
for(int i = 0; i < n; i++)
fin >> perm[i];
cout << minimum_walk(perm, s) << endl;
return 0;
}*/
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
384 KB |
Output is correct |
2 |
Correct |
2 ms |
384 KB |
Output is correct |
3 |
Correct |
2 ms |
384 KB |
Output is correct |
4 |
Correct |
2 ms |
384 KB |
Output is correct |
5 |
Correct |
2 ms |
384 KB |
Output is correct |
6 |
Correct |
2 ms |
384 KB |
Output is correct |
7 |
Correct |
2 ms |
384 KB |
Output is correct |
8 |
Correct |
2 ms |
384 KB |
Output is correct |
9 |
Correct |
2 ms |
384 KB |
Output is correct |
10 |
Correct |
2 ms |
356 KB |
Output is correct |
11 |
Correct |
2 ms |
384 KB |
Output is correct |
12 |
Correct |
2 ms |
384 KB |
Output is correct |
13 |
Correct |
2 ms |
384 KB |
Output is correct |
14 |
Correct |
2 ms |
412 KB |
Output is correct |
15 |
Correct |
2 ms |
384 KB |
Output is correct |
16 |
Correct |
2 ms |
384 KB |
Output is correct |
17 |
Correct |
2 ms |
384 KB |
Output is correct |
18 |
Correct |
3 ms |
384 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
384 KB |
Output is correct |
2 |
Correct |
2 ms |
384 KB |
Output is correct |
3 |
Correct |
2 ms |
384 KB |
Output is correct |
4 |
Correct |
2 ms |
384 KB |
Output is correct |
5 |
Correct |
2 ms |
384 KB |
Output is correct |
6 |
Correct |
2 ms |
384 KB |
Output is correct |
7 |
Correct |
2 ms |
384 KB |
Output is correct |
8 |
Correct |
2 ms |
384 KB |
Output is correct |
9 |
Correct |
2 ms |
384 KB |
Output is correct |
10 |
Correct |
2 ms |
356 KB |
Output is correct |
11 |
Correct |
2 ms |
384 KB |
Output is correct |
12 |
Correct |
2 ms |
384 KB |
Output is correct |
13 |
Correct |
2 ms |
384 KB |
Output is correct |
14 |
Correct |
2 ms |
412 KB |
Output is correct |
15 |
Correct |
2 ms |
384 KB |
Output is correct |
16 |
Correct |
2 ms |
384 KB |
Output is correct |
17 |
Correct |
2 ms |
384 KB |
Output is correct |
18 |
Correct |
3 ms |
384 KB |
Output is correct |
19 |
Correct |
3 ms |
256 KB |
Output is correct |
20 |
Correct |
2 ms |
356 KB |
Output is correct |
21 |
Correct |
2 ms |
384 KB |
Output is correct |
22 |
Correct |
2 ms |
384 KB |
Output is correct |
23 |
Correct |
2 ms |
384 KB |
Output is correct |
24 |
Correct |
2 ms |
384 KB |
Output is correct |
25 |
Correct |
3 ms |
384 KB |
Output is correct |
26 |
Correct |
2 ms |
384 KB |
Output is correct |
27 |
Correct |
2 ms |
384 KB |
Output is correct |
28 |
Correct |
3 ms |
384 KB |
Output is correct |
29 |
Correct |
3 ms |
384 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
384 KB |
Output is correct |
2 |
Correct |
2 ms |
384 KB |
Output is correct |
3 |
Correct |
2 ms |
384 KB |
Output is correct |
4 |
Correct |
2 ms |
384 KB |
Output is correct |
5 |
Correct |
2 ms |
384 KB |
Output is correct |
6 |
Correct |
2 ms |
384 KB |
Output is correct |
7 |
Correct |
2 ms |
384 KB |
Output is correct |
8 |
Correct |
2 ms |
384 KB |
Output is correct |
9 |
Correct |
2 ms |
384 KB |
Output is correct |
10 |
Correct |
2 ms |
356 KB |
Output is correct |
11 |
Correct |
2 ms |
384 KB |
Output is correct |
12 |
Correct |
2 ms |
384 KB |
Output is correct |
13 |
Correct |
2 ms |
384 KB |
Output is correct |
14 |
Correct |
2 ms |
412 KB |
Output is correct |
15 |
Correct |
2 ms |
384 KB |
Output is correct |
16 |
Correct |
2 ms |
384 KB |
Output is correct |
17 |
Correct |
2 ms |
384 KB |
Output is correct |
18 |
Correct |
3 ms |
384 KB |
Output is correct |
19 |
Correct |
3 ms |
256 KB |
Output is correct |
20 |
Correct |
2 ms |
356 KB |
Output is correct |
21 |
Correct |
2 ms |
384 KB |
Output is correct |
22 |
Correct |
2 ms |
384 KB |
Output is correct |
23 |
Correct |
2 ms |
384 KB |
Output is correct |
24 |
Correct |
2 ms |
384 KB |
Output is correct |
25 |
Correct |
3 ms |
384 KB |
Output is correct |
26 |
Correct |
2 ms |
384 KB |
Output is correct |
27 |
Correct |
2 ms |
384 KB |
Output is correct |
28 |
Correct |
3 ms |
384 KB |
Output is correct |
29 |
Correct |
3 ms |
384 KB |
Output is correct |
30 |
Correct |
288 ms |
35524 KB |
Output is correct |
31 |
Correct |
282 ms |
35532 KB |
Output is correct |
32 |
Correct |
122 ms |
19960 KB |
Output is correct |
33 |
Correct |
173 ms |
35648 KB |
Output is correct |
34 |
Correct |
184 ms |
35576 KB |
Output is correct |
35 |
Correct |
199 ms |
35576 KB |
Output is correct |
36 |
Correct |
180 ms |
35576 KB |
Output is correct |
37 |
Correct |
155 ms |
35576 KB |
Output is correct |
38 |
Correct |
147 ms |
35656 KB |
Output is correct |
39 |
Correct |
146 ms |
35536 KB |
Output is correct |
40 |
Correct |
157 ms |
35536 KB |
Output is correct |
41 |
Correct |
203 ms |
35548 KB |
Output is correct |
42 |
Correct |
235 ms |
35532 KB |
Output is correct |
43 |
Correct |
147 ms |
35600 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
384 KB |
Output is correct |
2 |
Correct |
2 ms |
384 KB |
Output is correct |
3 |
Incorrect |
2 ms |
384 KB |
3rd lines differ - on the 1st token, expected: '5920', found: '5986' |
4 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
384 KB |
Output is correct |
2 |
Correct |
2 ms |
384 KB |
Output is correct |
3 |
Correct |
2 ms |
384 KB |
Output is correct |
4 |
Correct |
2 ms |
384 KB |
Output is correct |
5 |
Correct |
2 ms |
384 KB |
Output is correct |
6 |
Correct |
2 ms |
384 KB |
Output is correct |
7 |
Correct |
2 ms |
384 KB |
Output is correct |
8 |
Correct |
2 ms |
384 KB |
Output is correct |
9 |
Correct |
2 ms |
384 KB |
Output is correct |
10 |
Correct |
2 ms |
356 KB |
Output is correct |
11 |
Correct |
2 ms |
384 KB |
Output is correct |
12 |
Correct |
2 ms |
384 KB |
Output is correct |
13 |
Correct |
2 ms |
384 KB |
Output is correct |
14 |
Correct |
2 ms |
412 KB |
Output is correct |
15 |
Correct |
2 ms |
384 KB |
Output is correct |
16 |
Correct |
2 ms |
384 KB |
Output is correct |
17 |
Correct |
2 ms |
384 KB |
Output is correct |
18 |
Correct |
3 ms |
384 KB |
Output is correct |
19 |
Correct |
3 ms |
256 KB |
Output is correct |
20 |
Correct |
2 ms |
356 KB |
Output is correct |
21 |
Correct |
2 ms |
384 KB |
Output is correct |
22 |
Correct |
2 ms |
384 KB |
Output is correct |
23 |
Correct |
2 ms |
384 KB |
Output is correct |
24 |
Correct |
2 ms |
384 KB |
Output is correct |
25 |
Correct |
3 ms |
384 KB |
Output is correct |
26 |
Correct |
2 ms |
384 KB |
Output is correct |
27 |
Correct |
2 ms |
384 KB |
Output is correct |
28 |
Correct |
3 ms |
384 KB |
Output is correct |
29 |
Correct |
3 ms |
384 KB |
Output is correct |
30 |
Correct |
288 ms |
35524 KB |
Output is correct |
31 |
Correct |
282 ms |
35532 KB |
Output is correct |
32 |
Correct |
122 ms |
19960 KB |
Output is correct |
33 |
Correct |
173 ms |
35648 KB |
Output is correct |
34 |
Correct |
184 ms |
35576 KB |
Output is correct |
35 |
Correct |
199 ms |
35576 KB |
Output is correct |
36 |
Correct |
180 ms |
35576 KB |
Output is correct |
37 |
Correct |
155 ms |
35576 KB |
Output is correct |
38 |
Correct |
147 ms |
35656 KB |
Output is correct |
39 |
Correct |
146 ms |
35536 KB |
Output is correct |
40 |
Correct |
157 ms |
35536 KB |
Output is correct |
41 |
Correct |
203 ms |
35548 KB |
Output is correct |
42 |
Correct |
235 ms |
35532 KB |
Output is correct |
43 |
Correct |
147 ms |
35600 KB |
Output is correct |
44 |
Correct |
2 ms |
384 KB |
Output is correct |
45 |
Correct |
2 ms |
384 KB |
Output is correct |
46 |
Incorrect |
2 ms |
384 KB |
3rd lines differ - on the 1st token, expected: '5920', found: '5986' |
47 |
Halted |
0 ms |
0 KB |
- |