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;
#define pb push_back
#define fi first
#define se second
#define sz(a) (int)(a.size())
#define all(a) a.begin(),a.end()
typedef long long int ll;
typedef pair<int,int> pii;
const int MAXN = 1e6+1;
int par[MAXN];
int find(int x){
if(x==par[x])return x;
return par[x] = find(par[x]);
}
ll minimum_walk(vector<int> p, int s) {
int n = sz(p);
ll ans = 0;
for(int i=0;i<n;i++){
ans+=abs(i-p[i]);
par[i] = i;
}
for(int i=0;i<n;i++){
if(i==p[i])continue;
int mn = min(i,p[i]);
int mx = max(i,p[i]);
int j = mn;
while(find(j)+1 <= mx){
j = find(j);
par[j] = j+1;
}
}
vector<int>lf,rg;
for(int i=0;i<n;i++){
if(i==p[i])continue;
int j = i;
while(j+1<n && find(i) == find(j+1))j++;
if(j<=s)lf.pb(j);
else if(i>=s)rg.pb(i);
else{
lf.pb(i);
rg.pb(j);
}
i = j;
}
if(!rg.empty())ans+=2*(rg.back()-s);
if(!lf.empty())ans+=2*(s-lf[0]);
return ans;
}
# | 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... |