商务旅行

商务旅行

题目描述 Description

某首都城市的商人要经常到各城镇去做生意，他们按自己的路线去做，目的是为了更好的节约时间。

假设有N个城镇，首都编号为1，商人从首都出发，其他各城镇之间都有道路连接，任意两个城镇之间如果有直连道路，在他们之间行驶需要花费单位时间。该国公路网络发达，从首都出发能到达任意一个城镇，并且公路网络不会存在环。

你的任务是帮助该商人计算一下他的最短旅行时间。

输入描述 Input Description

输入文件中的第一行有一个整数N，1<=n<=30 000，为城镇的数目。下面N-1行，每行由两个整数a 和b (1<=a, b<=n; a<>b)组成，表示城镇a和城镇b有公路连接。在第N+1行为一个整数M，下面的M行，每行有该商人需要顺次经过的各城镇编号。

输出描述 Output Description

    在输出文件中输出该商人旅行的最短时间。

样例输入 Sample Input

5

1 2

1 5

3 5

4 5

4

1

3

2

5

样例输出 Sample Output

7

#include<cstdio>
#include<vector>
#include<cmath>
using namespace std;
#define maxn 30009
vector<int>edge[maxn];
int dep[maxn], fa[maxn][20], len[maxn][20];
bool wasfa[maxn];
int ans=0;
void dfs(int cur)
{
	int k, i, son;
	dep[cur]=dep[fa[cur][0]]+1;
	k=ceil(log(dep[cur])/log(2));	//当前节点最多跳几次 
	for(i=1; i<=k; i++)
	{
		fa[cur][i]=fa[fa[cur][i-1]][i-1];
		len[cur][i]=len[cur][i-1]+len[fa[cur][i-1]][i-1];
	}
	vector<int>::iterator it;
	wasfa[cur]=1;
	for(it=edge[cur].begin(); it!=edge[cur].end(); it++)
	{
		son=(*it);
		if(!wasfa[son])
		{
			wasfa[son]=true;
			fa[son][0]=cur;		//这里容易漏写，这里不告诉儿子他爹是谁，他就再也不知道了（除非传参） 
			len[son][0]=1;		//相邻点的时间都是1 
			dfs(son);
		}
	}
}
void lca(int x, int y)
{
	int k, i;
	int s;
	if(dep[x]<dep[y])	swap(x, y);
	k=dep[x]-dep[y];
	s=ceil(log(dep[x])/log(2));
	for(i=0; i<=s; i++)		//这个地方我写错过，写成了从1开始循环，应该从父亲节点开始循环 
	{
		if(k&(1<<i))
		{
			ans+=len[x][i];
			x=fa[x][i];
		}
	}
	if(x==y)	return;
	s=ceil(log(dep[x])/log(2));
	for(i=s; i>=0; i--)
	{
		if(fa[x][i]!=fa[y][i])
		{
			ans+=len[x][i];
			x=fa[x][i];
			ans+=len[y][i];
			y=fa[y][i];
		}
	}
	ans+=2;//最后x,y点都在最近公共祖先的儿子位置处，所以要+2 
	return;
}
int main()
{
	int i, a, b, m, to, pre;
	int n;
	scanf("%d", &n);
	for(i=1; i<n; i++)
	{
		scanf("%d%d", &a, &b);
		edge[a].push_back(b);
		edge[b].push_back(a);
	}
	dfs(1);
	scanf("%d", &m);
	scanf("%d", &pre);
	for(i=1; i<m; i++)
	{
		scanf("%d", &to);
		lca(pre, to);
		pre=to;
	}
	printf("%d", ans);
	return 0;
}