LCA SP913 QTREE2 - Query on a tree II

SP913 QTREE2 - Query on a tree II

给定一棵n个点的树，边具有边权。要求作以下操作：

DIST a b 询问点a至点b路径上的边权之和

KTH a b k 询问点a至点b有向路径上的第k个点的编号

有多组测试数据，每组数据以DONE结尾。

裸的LCA。

在处理第二个操作时，我直接向上数跳了多少个。

顾z大佬说不能这么做，要求出跳到那个点的深度再去跳。

真的是这样，不过懒得想了，应该是+1-1的误差。 balabala。。。

code:

#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;

const int wx=50017;

int dep[wx],dis[wx];
int f[wx][23];
int head[wx];
int num,n,t;
char opt[7];

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,dis;
}edge[wx*2];

void add(int from,int to,int dis){
	edge[++num].nxt=head[from];
	edge[num].to=to;
	edge[num].dis=dis;
	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;dis[v]=dis[u]+edge[i].dis;
		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 find(int x,int k){
	for(int i=21;i>=0;i--){
		if(dep[f[x][i]]>=k)x=f[x][i];
	}
	return x;
}

int main(){
	t=read();
	while(t--){
		n=read();
		memset(head,0,sizeof head); num=1;
		memset(edge,0,sizeof edge);
		for(int i=1;i<n;i++){
			int x,y,z;
			x=read(); y=read(); z=read();
			add(x,y,z); add(y,x,z);
		}
		dfs(1,0); pre();
		while(1){
			scanf("%s",opt+1);
			if(opt[2]=='O')break;
			if(opt[2]=='I'){
				int x,y;
				x=read(); y=read();
				int lca=LCA(x,y);
				printf("%d\n",dis[x]+dis[y]-2*dis[lca]);
			}
			if(opt[1]=='K'){
				int a,b,k;
				a=read(); b=read(); k=read();
				int lca=LCA(a,b);
				if(dep[a]-dep[lca]+1>=k)printf("%d\n",find(a,dep[a]-k+1));
				else printf("%d\n",find(b,k-dep[a]+2*dep[lca]-1));
			}
		}
	}
}