| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 74370 | nvmdava | 고대 책들 (IOI17_books) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "books.h"
#include <bits/stdc++.h>
using namespace std;
int r = -1, l = 1000001;
vector<int> p;
bool in[1000001];
int cl = -1;
int s;
int cr = 1000001;
void go(int v){
if(in[v] == 1){
return;
}
if(v <= s){
cl = max(cl, v);
} else {
cr = min(cr, v);
}
in[v] = 1;
r = max(r, v);
go(p[v]);
}
long long minimum_walk(vector<int> p, int s) {
::p = p;
::s = s;
int q = 0;
long long ans = 0;
for(int i = 0; i < p.size(); i++){
ans += (long long)abs(i - p[i]);
if(in[i] || p[i] == i) continue;
l = min(l, i);
if(r < i && r != -1){
ans += (long long)(i - r) * 2;
}
go(i);
}
if(in[s] == 0){
if(cl != -1 && cr != 1000001) ans +=2 * min(s - cl, cr - s);
else if(cl != -1) ans += 2 * (s - cl);
else if(cr != 1000001) ans += 2 * (cr - s);
}
return ans;
}
int main(){
int n, s;
cin>>n>>s;
vector<int> p;
while(n--){
int k;
cin>>k;
p.push_back(k);
}
cout<<minimum_walk(p, s);
}