제출 #137650

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
137650		MAMBA	고대 책들 (IOI17_books)	C++17	0 / 100	2 ms	376 KiB

books

#include "books.h"
#include <bits/stdc++.h>

using namespace std;

#define rep(i , j , k) for(int i = j; i < (int)k; i++)

typedef long long ll;

const int N = 1e6 + 10;

int par[N];
ll le[N], ri[N], arr[N];

int getPar(int v) {
	return v == par[v] ? v : par[v] = getPar(par[v]); 
}

inline bool merge(int u , int v) {
	if ((u = getPar(u)) == (v = getPar(v))) 
		return false;
	par[v] = u;
	return true;
}

inline void smax(ll &a, ll b) {
	if (a < b) a = b;
}

inline void smin(ll &a, ll b) {
	if (b < a) a = b;
}

long long minimum_walk(std::vector<int> p, int s) {

	int n = p.size();

	iota(par , par + n , 0);

	ll res = 0;
	rep(i , 0 , n) {
		res += abs(p[i] - i);
		merge(p[i] , i);
	}


	fill(le , le + n , -1e9);
	fill(ri , ri + n , 1e9);

	rep(i , 0 , n) {
		if (i < s) 
			smax(le[getPar(i)] , i);
		else if (i > s)
			smin(ri[getPar(i)] , i);
	}

	ll Rlimit = s , Llimit = s;

	fill(arr ,arr + n , s);


	rep(i , 0 , n) { 
		if (par[i] == i && i != getPar(s) && i != p[i]) {
			if (le[i] < 0) smax(Rlimit , ri[i]);
			if (ri[i] >= n) smin(Llimit , le[i]);
			if (le[i] >= 0)
				arr[le[i]] = ri[i];
		}
	}



	ll ans = 1e18;
	rep(i , 0 , s + 1) {
		if (i <= Llimit) 
			smin(ans , res + (Rlimit - i) * 2);

		smax(Rlimit , arr[i]);
	}

	return ans;

}

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