Submission #138860

#TimeUsernameProblemLanguageResultExecution timeMemory
138860win11905Ancient Books (IOI17_books)C++11
50 / 100
328 ms20828 KiB
#include <bits/stdc++.h>
#include "books.h"
#define long long long 
using namespace std;

const int N = 1e6+5;

long ans = 0;
int n, x, y;
int id[N], L[N], R[N], ptr;
bool check[N];

void extend(int &l, int &r) {
	int ll = min(L[id[l]], L[id[r]]);
	int rr = max(R[id[l]], R[id[r]]);
	while(ll < l || r < rr) {
		if(ll < l) l--;
		else r++;
		ll = min(ll, min(L[id[l]], L[id[r]]));
		rr = max(rr, max(R[id[l]], R[id[r]]));
	}
}

long minimum_walk(vector<int> p, int s) {
	n = p.size();
	x = y = s;
	for(int i = 0; i < n; ++i) ans += abs(i - p[i]);
	for(int i = 0; i < n; ++i) if(!check[i]) {
		int u = i; ptr++;
		L[ptr] = R[ptr] = i;
		while(!check[u]) {
			id[u] = ptr;
			check[u] = true;
			R[ptr] = max(R[ptr], u);
            L[ptr] = min(L[ptr], u);
			u = p[u];
		}
		if(i != p[i]) x = min(x, i), y = max(y, i);
	}
	int l = s, r = s;
	while(x < l || r < y) {
		extend(l, r);
		int al = l, ar = r;
		int bl = l, br = r;
		long ca = 0, cb = 0;
		while(x < al && ar == r) ca += 2, extend(--al, ar);
		while(bl == l && br < y) cb += 2, extend(bl, ++br);
		if(ar != r && bl != l) ans += min(ca, cb);
		else ans += ca + cb;
		l = min(al, bl), r = max(ar, br);
	}
	return ans;
}
#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...