Submission #90585

#	Time	Username	Problem	Language	Result	Execution time	Memory
90585		mrtsima22	Birthday gift (IZhO18_treearray)	C++17	30 / 100	4093 ms	3560 KiB

treearray

#include <iostream>
#include <vector>
using namespace std;
int n,k,q,t_out[100003],fin,p[100003][22],deep[100003];
vector<int>a[100003];
int ba[100003],t_in[100004],T,x;
bool is_it(int a,int b)
{
	return (t_in[a]<=t_in[b])&&(t_out[b]<=t_out[a]);
}
void go(int s,int parent,int DEEP)
{
	t_in[s]=++T;
	p[s][1]=parent;
	deep[s]=DEEP;
	for(int i=0;i<(int)a[s].size();i++)
	{
		if(a[s][i]==parent)continue;
		go(a[s][i],s,DEEP+1);
	}
	t_out[s]=++T;
}
void mshobelize(int Par,int sad)
{
	for(int j=2;j<=19;j++)
	{
		p[sad][j]=p[p[sad][j-1]][j-1];
	}
	for(int i=0;i<(int)a[sad].size();i++)
	{
		if(a[sad][i]!=Par)
		{
			mshobelize(sad,a[sad][i]);
		}
	}
}
int lca(int a,int b)
{
	int A=deep[a];
	int B=deep[b];
	if(A>B)
	{
		swap(a,b);
		swap(A,B);
	}
	for(int i=19;i>=0;i--)
	{
		if(deep[p[b][i]]>=deep[a])
		{
			b=p[b][i];
		}
	}
	if(a==b)
	{
		return a;
	}
    for (int i = 19; i >= 0; i--) 
	{
        if (p[b][i] != p[a][i]) 
		{
            b = p[b][i];       
            a = p[a][i];
        }
    }
    return p[a][1];
}
int u,v,tp;
int main() {ios_base::sync_with_stdio(false);
cin>>n>>k>>q;
for(int i=0;i<n-1;i++)
{
	cin>>v>>u;
	a[v].push_back(u);
	a[u].push_back(v);
}
for(int i=1;i<=k;i++)
{
	cin>>ba[i];
}
go(1,0,0);
mshobelize(1,1);
t_out[0]=999999999;
while(q--)
{
	cin>>tp;
	if(tp==1)
	{
		cin>>u>>v;
		ba[u]=v;
	}
	else if(tp==2){
	cin>>u>>v>>x;
	fin=0;
		for(int L=u;!fin&&L<=v;L++)
		{
			int c=ba[L];
			int c1=c;
			for(int R=L;!fin&&R<=v;R++)
			{
				while(!is_it(c,ba[R]))c=p[c][1];
				c1=lca(c1,ba[R]);
//				if(c1!=c)
//					cout<<6/0<<endl;
				if(!is_it(x,c))break;
				if(x==c)
				{
					fin=1;
					cout<<L<<" "<<R<<endl;
				}
			}
		}
		if(!fin)
		{
			cout<<"-1 -1"<<endl;
		}
	}
	
}
 
 
	return 0;
}

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