LCA 【bzoj 4281】 [ONTAK2015]Związek Harcerstwa Bajtockiego

【bzoj 4281】 [ONTAK2015]Związek Harcerstwa Bajtockiego

Description

给定一棵有n个点的无根树，相邻的点之间的距离为1，一开始你位于m点。之后你将依次收到k个指令，每个指令包含两个整数d和t，你需要沿着最短路在t步之内（包含t步）走到d点，如果不能走到，则停在最后到达的那个点。请在每个指令之后输出你所在的位置。

Input

第一行包含三个正整数n,m,k(1<=m<=n<=1000000,1<=k<=1000000)。

接下来n-1行，每行包含两个正整数x,y(1<=x,y<=n)，描述一条树边。

接下来k行，每行两个整数d,t(1<=d<=n,0<=t<=10^9)，描述一条指令。

Output

输出一行，包含k个正整数，即执行每条指令后你所在的位置。

LCA简单题，写个函数模拟往上跳的过程就行。还要分情况讨论。

code：

#include<iostream>
#include<cstdio>
using namespace std;
const int wx=1000017;
inline int read(){
	int sum=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){sum=(sum<<1)+(sum<<3)+ch-'0';ch=getchar();}
	return sum*f;
}
struct e{
	int nxt,to;
}edge[wx*2];
int n,m,k,ans,x,y,z,now,num;
int head[wx],f[wx][23],dep[wx];
void add(int from,int to){
	edge[++num].nxt=head[from];
	edge[num].to=to;
	head[from]=num;
}
void dfs(int u,int fa){
	dep[u]=dep[fa]+1;
	for(int i=head[u];i;i=edge[i].nxt){
		int v=edge[i].to;
		if(v==fa)continue;
		f[v][0]=u;
		dfs(v,u);
	}
}
void pre(){
	for(int j=1;j<=21;j++){
		for(int i=1;i<=n;i++){
			f[i][j]=f[f[i][j-1]][j-1];
		}
	}
}
int LCA(int x,int y){
	if(dep[x]<dep[y])swap(x,y);
	for(int i=21;i>=0;i--){
		if(dep[f[x][i]]>=dep[y])x=f[x][i];
	}
	if(x==y)return x;
	for(int i=21;i>=0;i--){
		if(f[x][i]!=f[y][i]){
			x=f[x][i];y=f[y][i];
		}
	}
	return f[x][0];
}
int work(int now,int x){
	for(int i=21;i>=0;i--){
		if(dep[now]-dep[f[now][i]]<=x){
			x-=(dep[now]-dep[f[now][i]]);now=f[now][i];
		}
	}
	return now;
}
int main(){
	n=read();now=read();k=read();
	for(int i=1;i<n;i++){
		int x,y;
		x=read();y=read();
		add(x,y);add(y,x);
	}
	dfs(1,0);pre();
	for(int i=1;i<=k;i++){
		int dis,to;
		to=read();dis=read();
		int lca=LCA(now,to);
		if(dep[now]+dep[to]-2*dep[lca]<=dis){
			now=to;
			printf("%d\n",now);
		}
		else if(dep[now]-dep[lca]>=dis){
			now=work(now,dis);
			printf("%d\n",now);
		}
		else{
			now=work(to,dep[now]+dep[to]-2*dep[lca]-dis);
			printf("%d\n",now);
		}
	}
	return 0;
}