商务旅行_codevs1036_lca

Description

---

　　某首都城市的商人要经常到各城镇去做生意，他们按自己的路线去做，目的是为了更好的节约时间。
　　假设有N个城镇，首都编号为1，商人从首都出发，其他各城镇之间都有道路连接，任意两个城镇之间如果有直连道路，在他们之间行驶需要花费单位时间。该国公路网络发达，从首都出发能到达任意一个城镇，并且公路网络不会存在环。
　　你的任务是帮助该商人计算一下他的最短旅行时间。

Input

---

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

Output

---

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

Analysis

---

又见lca模mú板
先遍历一次求出每个点的深度，那么(a,b)间的距离就是dep[a]+dep[b]-2*dep[lca(a,b)]
最后输出距离和即可
多打几遍才有印象加深理解，这是真的，是真的

Code

---

#include <stdio.h>
#include <string.h>
using namespace std;
struct edge
{int y,next;}e[60001];
int ind[30001],dep[30001],ls[60001],f[30001][21],maxE=0;
void add(int x,int y)
{
    e[++maxE]=(edge){y,ls[x]};
    ls[x]=maxE;
}
void dfs(int fa,int x)
{
    f[x][0]=fa;
    for (int j=1;j<=20;j++)
        f[x][j]=f[f[x][j-1]][j-1];
    dep[x]=dep[fa]+1;
    for (int i=ls[x];i;i=e[i].next)
        if (!dep[e[i].y])
            dfs(x,e[i].y);
}
int LCA(int qx,int qy)
{
    if (dep[qx]<dep[qy])
        {qx^=qy;qy^=qx;qx^=qy;}
    for (int i=20;i>-1;i--)
        if (dep[qy]<=dep[f[qx][i]])
            qx=f[qx][i];
    if (qx==qy)
        return qx;
    for (int i=20;i>-1;i--)
        if (f[qx][i]!=f[qy][i])
        {
            qx=f[qx][i];
            qy=f[qy][i];
        }
    return f[qx][0];
}
int main()
{
    int n;
    scanf("%d",&n);
    for (int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        ind[y]++;
        add(x,y);
        add(y,x);
        f[y][0]=x;
        f[x][0]=y;
    }
    dfs(0,1);
    int m,now,last,ans=0;
    scanf("%d%d",&m,&last);
    for (int i=2;i<=m;i++)
    {
        scanf("%d",&now);
        ans+=dep[last]+dep[now]-dep[LCA(last,now)]*2;
        last=now;
    }
    printf("%d\n",ans);
    return 0;
}