spoj913 QTREE2 Query on a treeⅡ

题目链接：http://www.spoj.com/problems/QTREE2/en/

题目大意：

N个节点的树，边的编号为1~N-1，每条边有一个权值，要求模拟两种操作：
1：DIST a b :求点a到点b之间的距离
2：KTH a b k : 求从a出发到b遇到的第k个节点的编号

题解：

树剖的话，对于操作一就类模版询问，对于操作二就开两个数组分别记录从a起走到的点及从b起走到的点（直接跳（不会超时orz） 最后直接在数组找
倍增LCA的话就找a,b的LCA看看第k个点是在a到LCA的路径上还是LCA到b的路径上，倍增找点√

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define maxn 50100

struct node
{
	int x,y,c,next;
}a[maxn],e[maxn];int len,first[maxn];
struct trnode
{
	int l,r,lc,rc,c;
}tr[maxn];int trlen,z;
int son[maxn],fa[maxn],dep[maxn];
int ys[maxn],top[maxn],tot[maxn];
int mymax(int x,int y){return (x>y)?x:y;}
void ins(int x,int y,int c)
{
	len++;a[len].c=c;
	a[len].x=x;a[len].y=y;
	a[len].next=first[x];first[x]=len;
}
void dfs1(int x)
{
	son[x]=0;tot[x]=1;
	for (int k=first[x];k!=-1;k=a[k].next)
	{
		int y=a[k].y;
		if (y!=fa[x])
		{
			dep[y]=dep[x]+1;
			fa[y]=x;
			dfs1(y);
			if (tot[son[x]]<tot[y]) son[x]=y;
			tot[x]+=tot[y];
		}
	}
}
void dfs2(int x,int tp)
{
	ys[x]=++z;top[x]=tp;
	if (son[x]!=0) dfs2(son[x],tp);
	for (int k=first[x];k!=-1;k=a[k].next)
	{
		int y=a[k].y;
		if (y!=fa[x] && y!=son[x])
		 dfs2(y,y);
	}
}
void bt(int l,int r)
{
	++trlen;int now=trlen;
	tr[now].l=l;tr[now].r=r;
	tr[now].lc=tr[now].rc=-1;
	tr[now].c=0;
	if (l<r)
	{
		int mid=(l+r)>>1;
		tr[now].lc=trlen+1;bt(l,mid);
		tr[now].rc=trlen+1;bt(mid+1,r);
	}
}
void change(int now,int x,int k)
{
	if (tr[now].l==tr[now].r) {tr[now].c=k;return;}
	int mid=(tr[now].l+tr[now].r)>>1,lc=tr[now].lc,rc=tr[now].rc;
	if (x<=mid) change(lc,x,k);
	else change(rc,x,k);
	tr[now].c=tr[lc].c+tr[rc].c;
}
int fsum(int now,int l,int r)
{
	if (tr[now].l==l && tr[now].r==r) return tr[now].c;
	int lc=tr[now].lc,rc=tr[now].rc,mid=(tr[now].l+tr[now].r)>>1;
	if (r<=mid) return fsum(lc,l,r);
	else if (l>mid) return fsum(rc,l,r);
	else return fsum(lc,l,mid)+fsum(rc,mid+1,r);
}
int query(int x,int y)
{
	int ans=0,tx=top[x],ty=top[y];
	while (tx!=ty)
	{
		if (dep[tx]>dep[ty])
		{
			int tt=tx;tx=ty;ty=tt;
			tt=x;x=y;y=tt;
		}ans+=fsum(1,ys[ty],ys[y]);
		y=fa[ty];ty=top[y];
	}
	if (x==y) return ans;
	else
	{
		if (dep[x]>dep[y]) {int tt=x;x=y;y=tt;}
		return ans+fsum(1,ys[son[x]],ys[y]);
	}
}
int l1,l2,lt1[maxn],lt2[maxn];//两个记录的数组
int stp(int x,int y,int k)//就是跳啊跳orz
{
	l1=l2=0;
	while (x!=y)
	{
		if (dep[x]>dep[y])
		{
			lt1[++l1]=x;x=fa[x];
			if (k<=l1) return lt1[k];//如果还没到LCA就找到了直接返回
		}else {lt2[++l2]=y;y=fa[y];}
	}
	k-=l1;lt2[++l2]=x;
	return lt2[l2-k+1];//过了LCA的就算一下在另一条链的位置 返回
}
int main()
{
	//freopen("a.in","r",stdin);
	//freopen("a.out","w",stdout);
	int T,n,i,x,y,c,k;char s[20];
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d",&n);
		len=0;memset(first,-1,sizeof(first));
		for (i=1;i<n;i++)
		{
			scanf("%d%d%d",&x,&y,&c);
			ins(x,y,c);ins(y,x,c);
			e[i].x=x;e[i].y=y;e[i].c=c;
		}
		fa[1]=0;dep[1]=0;dfs1(1);
		z=0;dfs2(1,1);
		trlen=0;bt(1,n);
		for (i=1;i<n;i++)
		{
			if (dep[e[i].x]>dep[e[i].y])
			{
				int tt=e[i].x;e[i].x=e[i].y;e[i].y=tt;
			}
			change(1,ys[e[i].y],e[i].c);
		}
		while (1)
		{
			scanf("
%s",s);
			if (s[1]=='O') break;
			else if (s[0]=='D')
			{
				scanf("%d%d",&x,&y);
				printf("%d
",query(x,y));
			}else if (s[0]=='K')
			{
				scanf("%d%d%d",&x,&y,&k);
				printf("%d
",stp(x,y,k));
			}
		}
	}
	return 0;
}