SSL P1746 商务旅游

目录：

题目：

　　商务旅游 题目

分析：

　　题目跟我们说要求的是两点间的最短距离，我当然可以用dfs，但是，为了寻求更为高效的算法，我们采用了LCA算法
　　但有人就会问了，LCA不是用来求两点的最近公共祖先吗？当然，你没有记错，但我们只需一个公式就一样可以推出最短距离：$dep[i]+dep[j]-2*dep[LCA(i,j)]$
　　这个就是我们$i和j$的最短距离，dep为点的深度，而LCA则是用来求他们公共祖先是哪个点

思路：

　　　1.建出邻接表
　　　2.深搜求深度
　　　3.常规LCA(树上倍增)求解

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#define LL long long
using namespace std;
inline LL read() {
    LL d=0,f=1;char s=getchar();
    while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
    while(s>='0'&&s<='9'){d=d*10+s-'0';s=getchar();}
    return d*f;
}
struct edge{
    int to,next;
}e[60001];
int ls[60001],f[60001][17],dep[60001];
void dfs(int fa,int now)
{
    f[now][0]=fa;
    dep[now]=dep[fa]+1;
    for(int i=ls[now];i;i=e[i].next)
      if(!dep[e[i].to]) dfs(now,e[i].to);
}
int LCA(int l,int r)
{
    if(dep[l]<dep[r]) swap(l,r);
    for(int i=16;i>=0;i--)
      if(dep[r]<=dep[f[l][i]])
        l=f[l][i];
    if(l==r) return l;
    for(int i=16;i>=0;i--)
      if(f[l][i]!=f[r][i])
      {
        l=f[l][i];
        r=f[r][i];
      }
    return f[l][0];
}
int main()
{
    int n=read(),l=0;
    int a,b;
    for(int i=2;i<=n;i++)
    {
        a=read();b=read();
        e[++l]=(edge){b,ls[a]};
        ls[a]=l;
        e[++l]=(edge){a,ls[b]};
        ls[b]=l;
    }
    dfs(0,1);
    for(int j=1;j<=16;j++)
      for(int i=1;i<=n;i++)
        f[i][j]=f[f[i][j-1]][j-1];
    //因为我们有m个点，所以需要滚动求解，还要累加
    int now,m=read(),last=read(),ans=0;
    for(int i=2;i<=m;i++)
    {
        now=read();
        ans+=dep[last]+dep[now]-2*dep[LCA(last,now)];
        last=now;
    }
    printf("%d",ans);
    return 0;
}