oj.uz

답안 #865682

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
865682 2023-10-24T14:00:39 Z vjudge1 Birthday gift (IZhO18_treearray) C++17
0 / 100
 2 ms 12892 KB 
#include <bits/stdc++.h>
 
#define file(s) if (fopen(s".in", "r")) freopen(s".in", "r", stdin), freopen(s".out", "w", stdout)
#define optimus_prime ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0)
#define fxd(x) fixed << setprecision(x)
#define all(a) (a.begin() , a.end())
#define popcount(x) __builtin_popcount(x)
#define lwb lower_bound
#define upb upper_bound
#define dl double long
#define ll long long
#define pb push_back
#define sz() size()
#define F first
#define S second
 
using namespace std;
 
const ll N = 1e5+9;
const ll L = 18;

ll n , m , q , a[N] , tin[N] , tout[N] , timer , up[N][L];
vector <int> g[N];
set <int> st[N] , st1[N];

void calc(ll cur , ll p){
	tin[cur]=++timer;
	up[cur][0]=p;
	for (int i = 1 ; i <= L ; i++)up[cur][i]=up[up[cur][i-1]][i-1];
	for (auto to : g[cur]){
		if (to!=p)calc(to , cur);
	}
	tout[cur]=timer;
}

bool upper(ll a , ll b){
	return (tin[a]<=tin[b]&&tout[b]<=tout[a]);
}

ll lca(ll a , ll b){
	if (a && (!b || upper(a , b)))return a;
	if (b && (!a || upper(b , a)))return b;
	for (int i = L ; i >= 0 ; i--){
		if (!up[a][i])continue;	
		if (!upper(up[a][i] , b))a=up[a][i];
	}
	return up[a][0];
}

void solve(){
	cin >> n >> m >> q;
	for (int i = 1 ; i < n ; i++){
		ll u , v;
		cin >> u >> v;
		g[u].pb(v);
		g[v].pb(u);
	}
	calc(1 , 0);
	for (int i = 1 ; i <= m ; i++){
		cin >> a[i];
		st[a[i]].insert(i);
		if (i>1)st1[lca(a[i-1] , a[i])].insert(i-1);
	}
	while (q--){
		ll tp;
		cin >> tp;
		if (tp==1){
			ll pos , v;
			cin >> pos >> v;
			if (pos<m)st1[lca(a[pos] , a[pos+1])].erase(pos);
			if (pos>1)st1[lca(a[pos-1] , a[pos])].erase(pos-1);
			st[a[pos]].erase(pos);
			a[pos]=v;
			st[a[pos]].insert(pos);
			if (pos<m)st1[lca(a[pos] , a[pos+1])].insert(pos);
			if (pos>1)st1[lca(a[pos-1] , a[pos])].insert(pos-1);
		}
		else {
			ll l , r , v;
			cin >> l >> r >> v;
			if (st[v].lwb(l)!=st[v].end() && (*st[v].lwb(l))<=r){
				ll x=*(st[v].lwb(l));
				cout << x << " " << x << "\n";
			}
			else if (st1[v].lwb(l)!=st1[v].end() && (*st1[v].lwb(l))<r){
				ll x=*(st1[v].lwb(l));
				cout << x << " " << x+1 << "\n";
			}
			else cout << "-1 -1\n";
			
		}
	}
}
	
signed main() {
	optimus_prime;
	solve();
return 0;
}

Compilation message

treearray.cpp: In function 'void calc(long long int, long long int)':
treearray.cpp:29:42: warning: iteration 17 invokes undefined behavior [-Waggressive-loop-optimizations]
   29 |  for (int i = 1 ; i <= L ; i++)up[cur][i]=up[up[cur][i-1]][i-1];
      |                                ~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~
treearray.cpp:29:21: note: within this loop
   29 |  for (int i = 1 ; i <= L ; i++)up[cur][i]=up[up[cur][i-1]][i-1];
      |                   ~~^~~~

Subtask #1
0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 12888 KB n=5
2 Incorrect 2 ms 12892 KB Jury has the answer but participant has not
3 Halted 0 ms 0 KB -

Subtask #2
0 / 18.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 12888 KB n=5
2 Incorrect 2 ms 12892 KB Jury has the answer but participant has not
3 Halted 0 ms 0 KB -

Subtask #3
0 / 26.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 12888 KB n=5
2 Incorrect 2 ms 12892 KB Jury has the answer but participant has not
3 Halted 0 ms 0 KB -

Subtask #4
0 / 44.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 12888 KB n=5
2 Incorrect 2 ms 12892 KB Jury has the answer but participant has not
3 Halted 0 ms 0 KB -