codevs 1036 商务旅行 LCA 解题报告

题目描述 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

思路

LCA即可，网上看到有树链剖分的题解，感觉用不上啊，直接LCA然后预处理一下。

代码

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int N=30000+5;
int n,m,deep[N],head[N],p[N][20],dis[N];
int x,y,a[N],num,ans;
struct edge 
{
    int next,v,w;
};
edge ed[N*4];
void build(int u,int v) 
{
    ed[++num].v=v;
    ed[num].w=1;
    ed[num].next=head[u];
    head[u]=num;
}
void dfs(int now,int val) 
{
    dis[now]=val;
    for (int i=head[now];i!=-1;i=ed[i].next)
    {
        int go=ed[i].v;
        if (deep[go]==0) 
        {
            deep[go]=deep[now]+1;
            p[go][0]=now;
            dfs(go,val+ed[i].w);
        }
    }
}
void init() 
{
    for (int j=1;(1<<j)<=n;j++)
    for (int i=1;i<=n;i++)
    if (p[i][j-1]!=0) p[i][j]=p[p[i][j-1]][j-1];
}
int lca(int a,int b) 
{
    if (deep[a]<deep[b]) swap(a,b);
    int i;
    for (i=0;(1<<i)<=deep[a];i++);
    i--;
    for (int j=i; j>=0; j--)
    if (deep[a]-(1<<j)>=deep[b]) a=p[a][j];
    if (a==b) return a;
    for (int j=i; j>=0; j--)
    if (p[a][j]!=0&&p[a][j]!=p[b][j]) a=p[a][j],b=p[b][j];
    return p[a][0];
}
int main() 
{
    memset(head,-1,sizeof(head));
    scanf("%d",&n);
    for (int i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        build(x,y);
        build(y,x);
    }
    scanf("%d",&m);
    deep[1]=1;
    dfs(1,0);
    init();
    for (int i=1;i<=m;i++)
    scanf("%d",&a[i]);
    for (int i=2;i<=m;i++)
    {
        x=a[i-1],y=a[i];
        int find=lca(x,y);
        ans=ans+dis[x]+dis[y]-2*dis[find];
    }
    printf("%d",ans);
    return 0;
}