poj 1986 最近公共祖先 (lca 倍增)

第一次做这个认识lca，了解倍增模板的实际用法

《挑战程序设计》328页有详细解释。饿的不行了，有需要的私聊吧

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

#define cl(a,b) memset(a,b,sizeof(a))
#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)
{
    for(int i=0;i<G[u].size();i++)
    {
         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;
        for(int j=1;j<30;j++)
        {
             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);
    for(int i=0;i<30;i++)
    {
        if((dep[v]-dep[u])>>i&1)
            ans+=p[v][i],v=parent[v][i];
    }
    if(u==v)return ans;
    for(int i=30-1;i>=0;i--)//向上遍历i层，肯定不到30层就有公共祖先
    {
        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);
    for(int i=0;i<m;i++)
    {
        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;
}