Submission #1109220

#	Time	Username	Problem	Language	Result	Execution time	Memory
1109220		stdfloat	Birthday gift (IZhO18_treearray)	C++17	0 / 100	1 ms	508 KiB

treearray

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

vector<int> d;

vector<vector<int>> E, sp;

void dfs(int x, int p = -1) {
	sp[x][0] = p;
	if (p != -1) d[x] = d[p] + 1;
	for (auto i : E[x])
		if (i != p) dfs(i, x);
}

int lca(int x, int y) {
	if (d[x] < d[y]) swap(x, y);

	for (int i = 19; i >= 0; i--)
		if (d[x] - (1 << i) >= d[y]) x = sp[x][i];

	if (x == y) return x;

	for (int i = 19; i >= 0; i--) {
		if (sp[x][i] != sp[y][i]) {
			x = sp[x][i]; y = sp[y][i];
		}
	}

	return sp[x][0];
}

int main() {
	ios::sync_with_stdio(false); cin.tie(nullptr);

	int n, m, q;
	cin >> n >> m >> q;

	E.assign(n, {});
	for (int i = 1; i < n; i++) {
		int x, y;
		cin >> x >> y; x--; y--;

		E[x].push_back(y);
		E[y].push_back(x);
	}

	d.assign(n, 0);
	sp.assign(n, vector<int>(20, -1));
	dfs(0);

	for (int i = 1; i < 20; i++) {
		for (int j = 0; j < n; j++)
			if (sp[j][i - 1] != -1) sp[j][i] = sp[sp[j][i - 1]][i - 1];
	}

	vector<int> a(m);
	for (auto &i : a) {
		cin >> i; i--;
	}

	vector<set<pair<int, int>>> s(n);
	for (int i = 0; i + 1 < m; i++) {
		s[a[i]].insert({i, i});
		s[lca(a[i], a[i + 1])].insert({i, i + 1});
	}

	while (q--) {
		int tp;
		cin >> tp;

		if (tp == 1) {
			int ps, v;
			cin >> ps >> v; ps--; v--;
		
			s[a[ps]].erase({ps, ps});
			if (ps) s[lca(a[ps - 1], a[ps])].erase({ps - 1, ps});
			if (ps + 1 < m) s[lca(a[ps], a[ps + 1])].erase({ps, ps + 1});

			a[ps] = v;

			s[a[ps]].insert({ps, ps});
			if (ps) s[lca(a[ps - 1], a[ps])].insert({ps - 1, ps});
			if (ps + 1 < m) s[lca(a[ps], a[ps + 1])].insert({ps, ps + 1});
		}
		else {
			int l, r, v;
			cin >> l >> r >> v; l--; r--; v--;

			auto t = s[v].lower_bound({l, l});
			if (t == s[v].end() || (*t).second > r) cout << "-1 -1\n";
			else cout << (*t).first + 1 << ' ' << (*t).second + 1 << '\n';
		}
	}
}

• Subtask 1: 0 / 12
• Subtask 2: 0 / 18
• Subtask 3: 0 / 26
• Subtask 4: 0 / 44