BZOJ 1602: [Usaco2008 Oct]牧场行走

树上倍增lca

水题啊，练练手。

先一遍dfs搞出树和father[i][j]（意为第i个节点的2^j的祖先）以及distance[i]（意为i到根节点的距离）

答案就是distance[x]+distance[y]-2*distance[lca(x,y)]         //lca(x,y)为最近公共祖先

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#define g getchar()
using namespace std;
struct re{
	int v,w,next;
}ed[2016];
int n,q,e,ex,x,y,w;
int head[2016],f[2016][15],depth[2016],dis[2016];
inline int read(){
	int x=0,f=1;char ch=g;
	while(ch>'9'||ch<'0'){if(ch=='-')f=-1;ch=g;}
	while(ch<='9'&&ch>='0'){x=x*10+ch-'0';ch=g;}
	return x*f;
}
inline void ins(int x,int y,int w){
	ed[++e].v=y;ed[e].w=w;ed[e].next=head[x];head[x]=e;
	ed[++e].v=x;ed[e].w=w;ed[e].next=head[y];head[y]=e;
}
void dfs(int x){
	for(int i=1;i<=ex;++i){
		if(1<<i>depth[x])break;
		f[x][i]=f[f[x][i-1]][i-1];
	}                                      //倍增处理祖先
	for(int i=head[x];i;i=ed[i].next){
		if(f[x][0]==ed[i].v)continue;
		int v=ed[i].v;
		depth[v]=depth[x]+1;
		dis[v]=dis[x]+ed[i].w;
		f[v][0]=x;
		dfs(v);
	}
}
int lca(int x,int y){
	if(depth[x]<depth[y])swap(x,y);
	int d=depth[x]-depth[y];
	for(int i=0;i<=9;i++)if((1<<i)&d)x=f[x][i];
	for(int i=9;i>=0;i--) if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
    if(x==y)return x;
    else return f[x][0];
}                                                //求公共祖先
int main(){
	n=read();q=read();
	for(ex=10;1<<ex>n;--ex);
	for(int i=1;i<n;++i){
		x=read();y=read();w=read();ins(x,y,w);
		
	}
	dfs(1);
	for(int i=1;i<=q;++i){
		x=read();y=read();
		printf("%d\n",dis[x]+dis[y]-2*dis[lca(x,y)]);
	}
	return 0;
}

QωQ是不是挺容易的？其实spfa应该也可以但是没试过。。