Submission #1061487

#TimeUsernameProblemLanguageResultExecution timeMemory
1061487Zicrus고대 책들 (IOI17_books)C++17
0 / 100
0 ms348 KiB
#include <bits/stdc++.h>
#include "books.h"
using namespace std;

typedef long long ll;

ll minimum_walk(vector<int> p, int s) {
    ll n = p.size();
	ll res = 0;
    for (int i = 0; i < n; i++) res += abs(p[i] - i);

    vector<bool> vst(n);
    vector<vector<ll>> cyc;
    vector<ll> mxCyc;
    for (int i = 0; i < n; i++) {
        if (vst[i] || p[i] == i) continue;
        cyc.push_back(vector<ll>());
        mxCyc.push_back(i);
 
        ll cur = i;
        do {
            cyc.back().push_back(cur);
            mxCyc.back() = max(mxCyc.back(), cur);
            cur = p[cur];
            vst[cur] = true;
        } while (cur != i);
    }
    ll m = cyc.size();

    ll mx = 0;
    for (int i = 1; i < m; i++) {
        mx = max(mx, mxCyc[i-1]);
        res += 2*max(0ll, cyc[i][0] - mx);
    }
    mx = max(mx, mxCyc.back());
    if (s < cyc[0][0]) res += 2*(cyc[0][0] - s);
    if (s > mx) res += 2*(s - mx);

    return res;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...