Submission #1305794

#	Time	Username	Problem	Language	Result	Execution time	Memory
1305794		basic	Construction of Highway (JOI18_construction)	C++20	16 / 100	2094 ms	3976 KiB

construction

#pragma GCC optimize("Ofast,unroll-loops")
#include <bits/stdc++.h>
#define A(v) v.begin(), v.end()
using namespace std;
using ll = long long;
ll n, c[101010], a, b, p[101010], f[101010], r;
vector<ll> v;
int main() {
	cin.tie(0)->sync_with_stdio(0);
	cin >> n;
	for (int i = 1; i <= n; v.push_back(c[i++]))
		cin >> c[i];
	sort(A(v)), v.erase(unique(A(v)), v.end());
	for (int i = 1; i <= n; i++)
		c[i] = upper_bound(A(v), c[i]) - v.begin();
	for (int i = 1; i < n; p[b] = a, i++) {
		cin >> a >> b;
		fill(f, f + n + 1, r = 0);
		for (int i = a; i; i = p[i]) {
			for (int k = c[i] - 1; k > 0; k -= k & -k)
				r += f[k];
			for (int k = c[i]; k <= n; k += k & -k)
				f[k]++;
			c[i] = c[b];
		}
		cout << r << '\n';
	}
	return 0;
}

• Subtask 1: 7 / 7
• Subtask 2: 9 / 9
• Subtask 3: 0 / 84