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