【POJ1986】Distance Queries(lca的应用)

记录一个菜逼的成长。。

题目链接

题目大意:
给你n个点，m条边(是一颗树
q个查询。
每个查询给出两个点，求两点的距离。

感觉上树链剖分也可做，不难但是写起来麻烦。

这里用倍增lca来做。代码比较好写。(其他lca的写法也可以的
两点$x$,$y$的距离就是$x$到根的距离+$y$到根的距离-两倍的$lca(x,y)$到根的距离

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <vector>
#include <cstring>
using namespace std;
#define cl(a,b) memset(a,b,sizeof(a))
const int maxn = 100000 + 10;
struct Edge{
	int to,next,w;
}edge[maxn];
int head[maxn],cnt;
int deep[maxn],anc[maxn][20];
int n,m;
int dis[maxn];
void add(int u,int v,int w)
{
	edge[cnt].to = v;
	edge[cnt].next = head[u];
	edge[cnt].w = w;
	head[u] = cnt++;
}
void init()
{
	cl(head,-1);
	cl(deep,0);
	cl(anc,-1);
	cl(dis,0);
	cnt = 0;
}
void dfs(int u,int pre,int d)
{
	deep[u] = d;
	for( int i = head[u]; ~i; i = edge[i].next ){
		int v = edge[i].to;
		if(v == pre)continue;
		anc[v][0] = u;
		dis[v] = dis[u] + edge[i].w;
		dfs(v,u,d+1);
	}
}
void Init()
{
	for( int j = 1; (1<<j) <= n; j++ ){
		for( int i = 1; i <= n; i++ ){
			if(anc[i][j-1] != -1){
				anc[i][j] = anc[anc[i][j-1]][j-1];
			}
		}
	}
}
int lca(int a,int b)
{
	int i,j;
	if(deep[a] < deep[b])swap(a,b);
	for( i = 0; (1<<i) <= deep[a]; i++ );
	i--;
	
	for( j = i; j >= 0; j-- ){
		if(deep[a] - (1<<j) >= deep[b]){
			a = anc[a][j];
		}
	}
	if(a == b)return a;
	
	for( j = i; j >= 0; j-- ){
		if(anc[a][j] != -1 && anc[a][j] != anc[b][j]){
			a = anc[a][j];
			b = anc[b][j];
		}
	}
	return anc[a][0];
}
int main()
{
	while(~scanf("%d%d",&n,&m)){
		init();
		char dir[10];
		for( int i = 0; i < m; i++ ){
			int u,v,w;
			scanf("%d%d%d%s",&u,&v,&w,dir);
			add(u,v,w);
			add(v,u,w);
		}
		dfs(1,1,0);
		Init();
		int q;scanf("%d",&q);
		while(q--){
			int u,v;
			scanf("%d%d",&u,&v);
			printf("%d\n",dis[u] + dis[v] - 2 * dis[lca(u,v)]);
		}
	}
	return 0;
}