제출 #338147

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
338147		cki86201	고대 책들 (IOI17_books)	C++17	12 / 100	4 ms	4204 KiB

books

#include <bits/stdc++.h>
#include "books.h"
using namespace std;

#define Fi first
#define Se second
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
#define all(x) (x).begin(), (x).end()
#define pb push_back
#define szz(x) (int)(x).size()
#define rep(i, n) for(int i=0;i<(n);i++)
typedef tuple<int, int, int> t3;

int c[1000010], lv[1000010], rv[1000010];
ll minimum_walk(vector<int> p, int s) {
	int n = szz(p);
	ll ans = 0;
	rep(i, n) ans += abs(i - p[i]);
	rep(i, n) if(!c[i]) {
		int l = i, r = i;
		for(int j=p[i];j!=i;j=p[j]) l = min(l, j), r = max(r, j);
		c[i] = 1, lv[i] = l, rv[i] = r;
		for(int j=p[i];j!=i;j=p[j]) c[j] = 1, lv[j] = l, rv[j] = r;
	}
	memset(c, 0, sizeof c);
	int l = s, r = s;
	auto expend = [&]() {
		for(int x=r+1;x<n;x++) {
			if(lv[x] < l) {
				int r2 = x, l2 = -1;
				for(int i=p[r2];i!=r2;i=p[i]) {
					if(i < l) l2 = max(l2, i);
				}
				return pii(l2, r2);
			}
		}
		return pii(-1, -1);
	};
	vector <int> q; q.pb(s);
	auto push = [&](int x) {
		if(l > x) {
			for(int a=l-1;a>=x;a--) q.pb(a);
			l = x;
		}
		if(r < x) {
			for(int a=r+1;a<=x;a++) q.pb(a);
			r = x;
		}
	};
	while(1) {
		rep(i, szz(q)) {
			int t = q[i];
			if(c[t]) continue;
			c[t] = 1;
			for(int x=p[t];x!=t;x=p[x]) {
				push(x);
				c[x] = 1;
			}
		}
		auto [l2, r2] = expend();
		if(l2 == -1) break;
		q.clear();
		int vl = 0, vr = 0;
		for(int i=r;i<r2;i=rv[i+1], vr++);
		for(int i=l;i>l2;i=lv[i-1], vl++);
		ans += 2 * min(vl, vr);
		push(l2);
		push(r2);
	}
	int L = 0, R = n - 1;
	while(L < l && p[L] == L) ++L;
	while(r < R && p[R] == R) --R;
	for(int i=r;i<R;i=rv[i+1], ans+=2);
	for(int i=l;i>L;i=lv[i-1], ans+=2);
	return ans;
}

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