poj1986 Distance Queries（lca又是一道模版题）

题目链接：http://poj.org/problem?id=1986

题意：就是老问题求val[u]+val[v]-2*val[root]就行。还有这题没有给出不联通怎么输出那么题目给出的数据一定

是联通的。

 

题解：就是单纯的lca。

 

#include <iostream>
#include <cstring>
#include <vector>
#include <cstdio>
using namespace std;
const int M = 8e4 + 10;
vector<pair<int , int> >vc[M];
int p[M][20] , val[M] , deep[M];
void dfs(int pos , int pre , int d) {p[pos][0] = pre;deep[pos] = d;int len = vc[pos].size();for(int i = 0 ; i < len ; i++) {int u = vc[pos][i].first;if(u != pre) {val[u] += val[pos] + vc[pos][i].second;dfs(u , pos , d + 1);}}
}
void init(int n) {for(int i = 0 ; i < 18 ; i++) {for(int j = 1 ; j <= n ; j++) {if(p[j][i] == -1) {p[j][i + 1] = -1;}else {p[j][i + 1] = p[p[j][i]][i];}}}
}
int lca(int a , int b) {if(deep[a] < deep[b]) {swap(a , b);}int d = deep[a] - deep[b];for(int i = 0 ; i < 18 ; i++) {if(d & (1 << i)) {a = p[a][i];}}if(a == b) {return a;}for(int i = 17 ; i >= 0 ; i--) {if(p[a][i] != p[b][i] && p[a][i] != -1) {a = p[a][i];b = p[b][i];}}return p[a][0];
}
int main() {int n , m , u , v , w , k;char cp[10];while(scanf("%d%d" , &n , &m) != EOF) {for(int i = 1 ; i <= n ; i++) {val[i] = 0;vc[i].clear();}for(int i = 1 ; i < n ; i++) {scanf("%d%d%d%s" , &u , &v , &w , cp);vc[u].push_back(make_pair(v , w));vc[v].push_back(make_pair(u , w));}memset(p , -1 , sizeof(p));dfs(1 , -1 , 1);init(n);scanf("%d" , &k);while(k--) {scanf("%d%d" , &u , &v);printf("%d\n" , val[u] + val[v] - 2 * val[lca(u , v)]);}}return 0;
}