商务旅行
原创
©著作权归作者所有:来自51CTO博客作者背着代码的蜗牛的原创作品,请联系作者获取转载授权,否则将追究法律责任
商务旅行
题目描述:
某首都城市的商人要经常到各城镇去做生意,他们按自己的路线去做,目的是为了更好的节约时间。
假设有N个城镇,首都编号为1,商人从首都出发,其他各城镇之间都有道路连接,任意两个城镇之间如果有直连道路,在他们之间行驶需要花费单位时间。该国公路网络发达,从首都出发能到达任意一个城镇,并且公路网络不会存在环。
你的任务是帮助该商人计算一下他的最短旅行时间。
输入描述:
输入文件中的第一行有一个整数N,1<=n<=30 000,为城镇的数目。下面N-1行,每行由两个整数a 和b (1<=a, b<=n; a<>b)组成,表示城镇a和城镇b有公路连接。在第N+1行为一个整数M,下面的M行,每行有该商人需要顺次经过的各城镇编号。
输出描述:
在输出文件中输出该商人旅行的最短时间。
样例输入:
5
1 2
1 5
3 5
4 5
4
1
3
2
5
样例输出:
7
#include<iostream>
using namespace std;
const int maxn=30010;
struct node
{
int to;
int next;
}e[maxn*2];
int n,m,ans,tot,f[maxn][30],deep[maxn],head[maxn];
void edd_edge(int u,int v)
{
tot++;
e[tot].to=v;
e[tot].next=head[u];
head[u]=tot;
}
void 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;
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];
}
int lca(int a,int b)
{
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)
a=f[a][i];
d>>=1;
}
}
if(a==b)
return a;
for(int i=20;i>=0;i--)
if(f[a][i]!=f[b][i])
{
a=f[a][i];
b=f[b][i];
}
return f[a][0];
}
int main()
{
int x,y;
cin>>n;
for(int i=1;i<=n-1;i++)
{
cin>>x>>y;
edd_edge(x,y);
edd_edge(y,x);
}
f[1][0]=1;
deep[1]=0;
build(1);
init();
cin>>m;
int start=1,end;
for(int i=1;i<=m;i++)
{
cin>>end;
int d=lca(start,end);
ans+=deep[start]+deep[end]-2*deep[d];
start=end;
}
cout<<ans;
return 0;
}