poj 1986 Distance Queries (LCA 倍增)

题目链接
题意：给出一颗带权树，询问点对(u,v)路径上的取值和
可以使用 树链剖分做，是没有更新的查询很简单，今天学下倍增计算lca，使用倍增做一下，先dfs跑出树的每个节点的深度，以及每个节点的2的次幂的父亲节点，然后就暴力跑一跑，题目给出的树可能是不连通的

#include<stdio.h>
#include<cstring>
#include<vector>
using namespace std;

#define cl(a,b) memset(a,b,sizeof(a))
#define sz(x) ((x).size())
#define rep(i,a,b) for(int i=(a);i<(b);i++)
#define ll long long
#define pb push_back

const int maxn = 200000;

vector<pair<int,int> > G[maxn];

int p[maxn][30];
int parent[maxn][30];
int dep[maxn];
void dfs(int u,int fa){
    rep(i,0,sz(G[u])){
        int v=G[u][i].first;
        if(v==fa)continue;
        p[v][0]=G[u][i].second;
        parent[v][0]=u;
        dep[v]=dep[u]+1;
        rep(j,1,30){
            int t=parent[v][j-1];
            if(t==0)continue;
            p[v][j]=p[v][j-1]+p[t][j-1];
            parent[v][j]=parent[t][j-1];
        }
        dfs(v,u);
    }
}
int Lca(int u,int v){
    int ans=0;
    if(dep[u]>dep[v])swap(u,v);
    rep(i,0,30){
        if((dep[v]-dep[u])>>i&1)
            ans+=p[v][i],v=parent[v][i];
    }
    if(u==v) return ans;//return u;
    for(int i=30-1;i>=0;i--){
        if(parent[u][i]!=parent[v][i]){
            ans+=p[u][i]+p[v][i];
            u=parent[u][i];
            v=parent[v][i];
        }
    }
    if(parent[u][0]==0||parent[v][0]==0)return 0;//给出的图可能是不连通的需要特判一下
    return ans+p[v][0]+p[u][0];// return parent[u][0];
}
int n,m,x,y,w;
char op[2];

int main(){
    scanf("%d%d",&n,&m);
    rep(i,0,m){
        scanf("%d%d%d%s",&x,&y,&w,op);
        G[x].pb(make_pair(y,w));
        G[y].pb(make_pair(x,w));
    }
    dfs(1,1);

    int k;scanf("%d",&k);
    while(k--){
        int x,y;scanf("%d%d",&x,&y);
        printf("%d\n",Lca(x,y));
    }

    return 0;
}