小机房的树
算法使用:
LCA
题目描述:
小机房有棵焕狗种的树,树上有N个节点,节点标号为0到N-1,有两只虫子名叫飘狗和大吉狗,分居在两个不同的节点上。有一天,他们想爬到一个节点上去搞基,但是作为两只虫子,他们不想花费太多精力。已知从某个节点爬到其父亲节点要花费 c 的能量(从父亲节点爬到此节点也相同),他们想找出一条花费精力最短的路,以使得搞基的时候精力旺盛,他们找到你要你设计一个程序来找到这条路,要求你告诉他们最少需要花费多少精力
输入描述 Input Description
第一行一个n,接下来n-1行每一行有三个整数u,v, c 。表示节点 u 爬到节点 v 需要花费 c 的精力。
第n+1行有一个整数m表示有m次询问。接下来m行每一行有两个整数 u ,v 表示两只虫子所在的节点
输出描述:
一共有m行,每一行一个整数,表示对于该次询问所得出的最短距离。
样例输入:
3
1 0 1
2 0 1
3
1 0
2 0
1 2
样例输出:
1
1
2
数据范围及提示:
1<=n<=50000, 1<=m<=75000, 0<=c<=1000
#include<iostream>
#include<cstdio>
using namespace std;
const int maxn=50010;
struct node
{
int to;
int w;
int next;
}e[maxn*10];
int n,m,tot,dis[maxn][25],deep[maxn],f[maxn][30],head[maxn];
void edd_edge(int u,int v,int w)
{
tot++;
e[tot].to=v;
e[tot].w=w;
e[tot].next=head[u];
head[u]=tot;
}
int build(int u)
{
for(int i=head[u];i;i=e[i].next)
if(!f[e[i].to][0])
{
f[e[i].to][0]=u;
deep[e[i].to]=deep[u]+1;
dis[e[i].to][0]=e[i].w;
build(e[i].to);
}
}
void init()
{
for(int j=1;j<=20;j++)
for(int i=1;i<=n;i++)
{
f[i][j]=f[f[i][j-1]][j-1];
dis[i][j]=dis[i][j-1]+dis[f[i][j-1]][j-1];
}
}
int lca(int a,int b)
{
int ans=0;
if(deep[a]<deep[b])
swap(a,b);
if(deep[a]!=deep[b])
{
int d=deep[a]-deep[b];
for(int i=0;d;i++)
{
if(d&1)
{
ans+=dis[a][i];
a=f[a][i];
}
d>>=1;
}
}
if(a==b)
return ans;
for(int i=20;i>=0;i--)
if(f[a][i]!=f[b][i])
{
ans+=dis[a][i];
ans+=dis[b][i];
a=f[a][i];
b=f[b][i];
}
ans+=dis[a][0];
ans+=dis[b][0];
return ans;
}
int main()
{
int x,y,z;
scanf("%d",&n);
for(int i=1;i<=n-1;i++)
{
scanf("%d%d%d",&x,&y,&z);
edd_edge(x+1,y+1,z);
edd_edge(y+1,x+1,z);
}
f[1][0]=1;
build(1);
init();
int start,end;
scanf("%d",&m);
for(int i=1;i<=m;i++)
{
scanf("%d%d",&start,&end);
int tmp=lca(start+1,end+1);
printf("%d\n",tmp);
}
return 0;
}