Submission #767662

#	Time	Username	Problem	Language	Result	Execution time	Memory
767662		SanguineChameleon	Ancient Books (IOI17_books)	C++17	0 / 100	1 ms	212 KiB

books

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

const int maxn = 1e6 + 20;
int par[maxn];
int depth[maxn];

int root(int u) {
	if (par[u] == -1) {
		return u;
	}
	else {
		return par[u] = root(par[u]);
	}
}

bool update(int u, int v) {
	int ru = root(u);
	int rv = root(v);
	if (ru == rv) {
		return false;
	}
	if (depth[ru] > depth[rv]) {
		swap(ru, rv);
	}
	par[ru] = rv;
	if (depth[ru] == depth[rv]) {
		depth[rv]++;
	}
	return true;
}

long long minimum_walk(vector<int> a, int s) {
	int n = a.size();
	for (int i = 0; i < n; i++) {
		par[i] = -1;
	}
	long long res = 0;
	for (int i = 0; i < n; i++) {
		res += abs(a[i] - i);
		update(i, a[i]);
	}
	for (int i = 0; i < n - 1; i++) {
		if (update(i, i + 1)) {
			res += 2;
		}
	}
	return res;
}

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