This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include "books.h"
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef vector<int>vi;
#define pb push_back
#define all(x) begin(x), end(x)
#define sz(x) (int)x.size()
#define FOR(i,a,b) for(int i=a; i<b; i++)
#define ROF(i,a,b) for(int i=b-1; i>=a; i--)
void dbgv(vi vec){for(int x: vec) cout << x << ' '; cout << endl; }
//---------------------------
int N;
vi a,p;
int check(){
FOR(i,0,N) if(a[i]==-1 || p[a[i]]!=i) return 0;
return 1;
}
int dist(int i, int j){
return abs(i-j);
}
ll minimum_walk(vi p, int s) {
::p=p;
N=sz(p);
a.resize(N);
iota(all(a),0);
int l=0, r=N-1, on=-1;
ll ans=0;
while(!check()){
while(p[a[l]]==l) l++, ans++;
while(p[a[r]]==r) r--;
//if(l>r) break;
FOR(i,l,r+1){
if(p[a[i]]>i){
if(on==-1 || p[a[i]]>p[on]) swap(a[i], on);
}
else if(on!=-1 && i==p[on]){
assert(i==r);
swap(a[i], on);
}
}
ROF(i,l,r+1){
if(a[i]==-1){
assert(i==l);
swap(a[i], on);
break;
}
if(p[a[i]]<i){
if(p[a[i]]<p[on]) swap(a[i], on);
}
}
ans+=(r-l)*2;
}
ans+=l;
return ans;
}
/*
4 0
0 2 3 1
*/
/*
3 0
2 0 1
*/
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |