제출 #137691

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

books

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

using namespace std;

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

	int last = -1;

	vector<pair<ll , pair<int , int>>> vec;
	rep(i , 0 , n) {
		if (i == p[i]) continue;
		if (last == -1 || getPar(last) == getPar(i))
			last = i;
		else 
			vec.pb({2 * (i - last) , {last , i}});
	}

	sort(vec.begin(), vec.end());

	for (auto e : vec)
		if (merge(e.second.first , e.second.second))
			res += e.first;
	return res;

}

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