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_backconst 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;}