Submission #245678

#	Time	Username	Problem	Language	Result	Execution time	Memory
245678		Yera	Birthday gift (IZhO18_treearray)	C++17	100 / 100	916 ms	141204 KiB

treearray

// In The Name Of God
//#pragma GCC optimize("Ofast,no-stack-protector,unroll-loops,fast-math,O3")
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
#include <iostream>
#include <iomanip>
#include <fstream>
#include <vector>
#include <set>
#include <map>
#include <cstring>
#include <string>
#include <bitset>
#include <cmath>
#include <cassert>
#include <ctime>
#include <algorithm>
#include <sstream>
#include <list>
#include <queue>
#include <deque>
#include <stack>
#include <cstdlib>
#include <cstdio>
#include <iterator>
#include <functional>
#include <unordered_set>
#include <unordered_map>
using namespace std;
using ll = long long;
using ld = long double;
using ull = unsigned long long;
 
#define f first
#define s second
#define pb push_back
#define mp make_pair
#define sagyndym_seni ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0)
#define sz(x) (int)x.size()
#define all(x) x.begin(), x.end()
 
const int N = 2e5+5, M = (1 << 20), p1 = 911382323, p2 = 972663749, INF = 1e9+123;
 
inline int read(){
	char c = getchar_unlocked();
	bool minus = 0;
	while (c < '0' || '9' < c){
		if(c == '-'){ minus = 1;}
		c = getchar_unlocked();
		if(c == '-'){ minus = 1;}
	}
	int res = 0;
	while ('0' <= c && c <= '9') {
		res = (res << 3) + (res << 1) + c - '0';
		c = getchar_unlocked();
	}
	if(minus){ res = (~res) + 1;}
	return res;
}
 
int n, m, q;
int pos[M], LOG[M], vec[N];
pair<int, int> st[M][20];
vector<int> g[N];
vector<pair<int, int>> order;
 
void dfs(int v, int p, int depth){
	order.pb({depth, v});
	assert(sz(order) < (N << 2));
	pos[v] = sz(order)-1;
	for(auto to : g[v]){
		if(to == p){ continue;}
		dfs(to, v, depth + 1);
		order.pb({depth, v});
	}
}
 
inline void precalc(){
	LOG[1] = 0;
	for(int i = 2; i < (N << 2); i++){
		LOG[i] = LOG[i >> 1] + 1;
	}
	for(int i = 0; i < sz(order); i++){
		assert(i < (N << 2));
		st[i][0] = order[i];
	}
	for(int j = 1; j < 30; j++){
		for(int i = 0; i + (1 << j) <= sz(order); i++){
			st[i][j] = min(st[i][j-1], st[i + (1 << (j-1))][j-1]);
		}
	}
}
 
inline int mn(int l, int r){
	if(r < l){ swap(l, r);}
	assert(r - l + 1 < (N << 2) && r < (N << 2));
	int j = LOG[r - l + 1];
	pair<int, int> res = min(st[l][j], st[r - (1 << j) + 1][j]);
	return res.s;
}

set<int> uno[N], duo[N];

int main(){
	sagyndym_seni;
	n = read(); m = read(); q = read();
	for(int i = 1; i < n; i++){
		int v = read()-1, u = read()-1;
		g[v].pb(u);
		g[u].pb(v);
	}
	for(int i = 0; i < m; i++){
		vec[i] = read()-1;
		uno[vec[i]].insert(i);
	}
	dfs(0, 0, 0);
	precalc();
	for(int i = 0; i < m - 1; i++){
		duo[mn(pos[vec[i]], pos[vec[i + 1]])].insert(i);
	}
	while(q--){
		int tp = read()-1;
		if(!tp){
			int p = read()-1, v = read()-1;
			uno[vec[p]].erase(p);
			uno[v].insert(p);
			if(p != m - 1){
				duo[mn(pos[vec[p]], pos[vec[p + 1]])].erase(p);
				duo[mn(pos[v], pos[vec[p + 1]])].insert(p);
			}
			if(p != 0){
				duo[mn(pos[vec[p - 1]], pos[vec[p]])].erase(p - 1);
				duo[mn(pos[vec[p - 1]], pos[v])].insert(p - 1);
			}
			vec[p] = v;
		}else{
			int l = read()-1, r = read()-1, p = read()-1;
			auto it = uno[p].lower_bound(l);
			if(it != uno[p].end()){
				int res = *it;
				if(l <= res && res <= r){ cout<<'\n'<<res + 1<<' '<<res+1; continue;}
			}
			it = duo[p].lower_bound(l);
			if(it != duo[p].end()){
				int res = *it;
				if(l <= res && res < r){ cout<<'\n'<<res + 1<<' '<<res + 2; continue;}
			}
			cout<<"\n-1 -1";
		}
	}
	return 0;
}
/* TIMUS: 292220YC*/

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