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;
const int N=1e6;
int vis[N], n;
long long minimum_walk(vector<int> p, int s) {
n=p.size();
vector<pair<int, int>> v;
long long ans=0;
vector<int> vv;
for (int i=0; i<n; ++i) if (!vis[i]){
int mx=i, mn=i;
while (!vis[i]){
if (i!=p[i]) vv.push_back(i);
mx=max(mx, i);
mn=min(mn, i);
vis[i]=1;
ans+=abs(p[i]-i);
i=p[i];
}
v.emplace_back(mn, mx);
}
int pf=0, sf=0;
for (int i=0; i<n; ++i) if (p[i]==i) ++pf; else break;
for (int i=n-1; i>=0; --i) if (p[i]==i) ++sf; else break;
sort(v.begin(), v.end());
int cur=0;
for (auto &i:v){
if (cur<=i.first) ans+=(i.first-cur)*2;
cur=max(cur, i.second);
}
bool check=0;
for (auto &i:v) check|=i.first<=s && s<=i.second;
if (pf>=s) ans-=s*2;
else ans-=pf*2;
if (sf>=n-s-1) ans-=(n-s-1)*2;
else ans-=sf*2;
sort(vv.begin(), vv.end());
if (check){
auto it=lower_bound(vv.begin(), vv.end(), s);
if (it==vv.begin()) ans+=(*it-s)*2;
else ans+=min(*it-s, s-*prev(it))*2;
}
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... |