| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 960950 | MarwenElarbi | 고대 책들 (IOI17_books) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#include "books.h"
using namespace std;
long long minimum_walk(std::vector<int> p, int s){
int n=p.size();
long long ans=0;
bool vis[n];
vector<int> cycles(n);
memset(vis,0,sizeof vis);
for (int i = 0; i < n; ++i)
{
ans+=abs(p[i]-i);
}
int en=n-1;
for (int i = n-1; i >= 0; --i)
{
en=i;
if(p[i]!=i) break;
}
//cout << ans<<endl;
for (int i = 0; i < n; ++i)
{
int cur=i;
if(vis[cur]) continue;
int mn=1e6+5;
int mx=-1;
while(vis[cur]==0){
vis[cur]=1;
mx=max(mx,cur);
mn=min(mn,cur);
cur=p[cur];
}
cycles[mn]++;
cycles[mx]--;
}
int cur=0;
for (int i = 0; i < en; ++i)
{
cur+=cycles[i];
if(cur==0) ans+=2;
}
return ans;
}
int main() {
#ifndef ONLINE_JUDGE
freopen("input.txt", "r", stdin);
freopen("output.txt", "w", stdout);
#endif
int n, s;
assert(2 == scanf("%d %d", &n, &s));
vector<int> p((unsigned) n);
for(int i = 0; i < n; i++)
assert(1 == scanf("%d", &p[(unsigned) i]));
long long res = minimum_walk(p, s);
printf("%lld\n", res);
return 0;
}