http://codevs.cn/problem/2370/
倍增LCA版本
代码
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
int n,m,u,v,c,tot;
int next[50001*2],first[50001],deep[50001],rank[50001],anc[50001][25];
struct edge
{
int t;
int d;
}l[100001];
void build(int u,int v,int c)
{
l[++tot].t=v;
l[tot].d=c;
next[tot]=first[u];
first[u]=tot;
}
void dfs(int u,int fa,int w)
{
deep[u]=deep[fa]+1;
anc[u][0]=fa;
rank[u]=w;
for(int i=1;anc[u][i-1];i++)
{
anc[u][i]=anc[anc[u][i-1]][i-1];
}
for(int i=first[u];i!=-1;i=next[i])//for(int i=first[u];i;i=next[i])i不为0
{
int v=l[i].t;
if(!deep[v])
{
dfs(v,u,w+l[i].d);
}
}
}
int ask_lca(int x,int y)
{
if(deep[x]<deep[y]) swap(x,y);
if(deep[x]>deep[y])
{
int dd=deep[x]-deep[y];
for(int i=0;i<=20;i++)
{
if(dd&(1<<i))
{
x=anc[x][i];
}
}
}
if(x!=y)
{
for(int i=20;i>=0;i--)
{
if(anc[x][i]!=anc[y][i])
{
x=anc[x][i];
y=anc[y][i];
}
}
}
if(x==y) return x;
else return anc[x][0];
}
int main()
{
memset(first,-1,sizeof(first));
scanf("%d",&n);
for(int i=1;i<=n-1;i++)
{
scanf("%d%d%d",&u,&v,&c);
build(u,v,c);
build(v,u,c);
}
dfs(0,0,0);
scanf("%d",&m);
for(int i=1;i<=m;i++)
{
scanf("%d%d",&u,&v);
printf("%d\n",rank[u]+rank[v]-(2*rank[ask_lca(u,v)]));
}
return 0;
}
树链剖分版本
代码
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
typedef long long ll;
int n,m,ru,rv,rw,a,b,tot;
int first[1000010],nxt[1000010],deep[1000010],f[1000010],top[1000010],siz[1000010],son[1000010],rank[1000010];
struct edge
{
int u,v,w;
}l[1000010];
void add(int f,int t,int c)
{
l[++tot]=(edge){f,t,c};
nxt[tot]=first[f];
first[f]=tot;
}
void dfs_1(int k,int fa,int d)
{
deep[k]=d;
f[k]=fa;
siz[k]=1;
for(int i=first[k];i!=-1;i=nxt[i])
{
int x=l[i].v;
if(x==fa)
continue;
dfs_1(x,k,d+1);
siz[k]+=siz[x];
if(!son[k]||siz[x]>siz[son[k]])
son[k]=x;
}
}
void dfs_2(int k,int num)
{
top[k]=num;
if(!son[k])
return;
dfs_2(son[k],num);
for(int i=first[k];i!=-1;i=nxt[i])
{
int x=l[i].v;
if(x!=son[k]&&x!=f[k])
dfs_2(x,x);
}
}
void dfs(int k,int d)
{
rank[k]=d;
for(int i=first[k];i!=-1;i=nxt[i])
{
int x=l[i].v;
if(x!=f[k])
dfs(x,d+l[i].w);
}
}
ll find_lca(int x,int y)
{
int f1=top[x],f2=top[y];
ll ans=0;
while(f1!=f2)
{
if(deep[f1]<deep[f2])
{
f1=f1^f2;
f2=f1^f2;
f1=f1^f2;
x=x^y;
y=x^y;
x=x^y;
}
ans+=rank[x]-rank[f[f1]];//注意边界问题
x=f[f1];
f1=top[x];
}
if(deep[x]>deep[y])
ans+=rank[x]-rank[y];
if(deep[y]>=deep[x])
ans+=rank[y]-rank[x];
return ans;
}
int main()
{
memset(first,-1,sizeof(first));
scanf("%d",&n);
for(int i=1;i<n;i++)
{
scanf("%d%d%d",&ru,&rv,&rw);
add(ru,rv,rw);
add(rv,ru,rw);
}
tot=0;
dfs_1(0,0,1);
dfs_2(0,0);
dfs(0,0);
scanf("%d",&m);
for(int i=1;i<=m;i++)
{
scanf("%d%d",&a,&b);
printf("%lld\n",find_lca(a,b));
}
return 0;
}