Submission #1061489

#	Time	Username	Problem	Language	Result	Execution time	Memory
1061489		Zicrus	Ancient Books (IOI17_books)	C++17	50 / 100	125 ms	27040 KiB

books

#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);
    }

    if (m == 0) return res;
    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;
}

• Subtask 1: 12 / 12
• Subtask 2: 10 / 10
• Subtask 3: 28 / 28
• Subtask 4: 0 / 20
• Subtask 5: 0 / 30