HDU 2586 How far away LCA 倍增法

倍增法求lca，关键就是在于预处理。

如果说题的数据太水，有时候暴力反而更快……

（此题倍增法我用了78ms，当年暴力62ms...）

参考资料：《挑战程序设计竞赛》P328

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#define N 40010
#define M 80010
#define exp 1e-7
using namespace std;

int T,n,m;
int u[M],v[M],w[M],fst[N],nxt[M],tot;
int dis[N],dep[N];
int f[N][20];

void init()
{
	memset(u,0,sizeof(u));
	memset(v,0,sizeof(v));
	memset(w,0,sizeof(w));
	memset(fst,0,sizeof(fst));
	memset(nxt,0,sizeof(nxt));
	tot=0;
	memset(dis,0,sizeof(dis));
	memset(dep,0,sizeof(dep));
	memset(f,0,sizeof(f));
}

void add_edge(int a,int b,int val)
{
	tot++;
	u[tot]=a; v[tot]=b; w[tot]=val;
	nxt[tot]=fst[a]; fst[a]=tot;
	tot++;
	u[tot]=b; v[tot]=a; w[tot]=val;
	nxt[tot]=fst[b]; fst[b]=tot;
}

inline int log2(int x)
{
	return (int)(log(x)/log(2)+exp);
}

void dfs(int p,int fa,int step,int depth)
{
	dis[p]=step; dep[p]=depth;
	f[p][0]=fa;
	for (int i=1;i<=log2(dep[p]);i++)
	{
		f[p][i]=f[f[p][i-1]][i-1];
	}
	for (int i=fst[p];i!=0;i=nxt[i])
	{
		if (v[i]!=fa)
		{
			dfs(v[i],p,step+w[i],depth+1);
		}
	}
}

int lca(int a,int b)
{
	if (dep[a]<dep[b]) swap(a,b);
	for (int i=20;i>=0;i--)
	{
		//printf("%d\n",a);
		if (dep[a]==dep[b]) break;
		if (dep[f[a][i]]<dep[b]) continue;
		a=f[a][i];
	}
	if (a==b) return a;
	for (int i=log2(dep[a]);i>=0;i--)
	{
		if (f[a][i]==f[b][i]) continue;
		a=f[a][i]; b=f[b][i];
	}
	return f[a][0];
}

int main()
{
	scanf("%d",&T);
	for (int tt=1;tt<=T;tt++)
	{
		init();
		scanf("%d%d",&n,&m);
		int x,y,z;
		for(int i=1;i<n;i++)
		{
			scanf("%d%d%d",&x,&y,&z);
			add_edge(x,y,z);
		}
		dfs(1,-1,0,0);
		for (int i=1;i<=m;i++)
		{
			scanf("%d%d",&x,&y);
			//printf("%d ",lca(x,y));
			printf("%d\n",dis[x]+dis[y]-2*dis[lca(x,y)]);
		}
	}
	return 0;
}