C++之路进阶——codevs1036（商务旅行）

1036 商务旅行 

题目描述 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。。。。，模板题。不懂看代码。

#include<cstdio>
#include<iostream>
#include<cmath>
#include<vector>
#include<algorithm>
#define maxn 30010

using namespace std;

vector<int> s[30010];
int fa[maxn][100],deep[maxn],n,m;

void  getdeep(int k,int d)
    {
        deep[k]=d;
        for (int i=1;(1<<i)<=deep[k];i++)
          fa[k][i]=fa[fa[k][i-1]][i-1];
        for (int i=0;i<s[k].size();i++)
          {
              if (!deep[s[k][i]])
                 {
                      fa[s[k][i]][0]=k;
                      getdeep(s[k][i],d+1);
                 }
          } 
    }
    
int lca(int x,int y)
  {
      if (deep[x]<deep[y]) swap(x,y);
      int t=deep[x]-deep[y];
      for (int i=0;i<20;i++)
       if ((1<<i)&t) x=fa[x][i];      
      for (int i=20;i>=0;i--)
         {
               if (fa[x][i]!=fa[y][i])
                   {
                         x=fa[x][i];
                         y=fa[y][i];
                   }
         } 
    if (x==y)    return x;
    return fa[x][0]; 
  }
  
int main()
  {
      scanf("%d",&n);
      for (int i=0;i<n-1;i++)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            s[y].push_back(x);
            s[x].push_back(y);
        }
    getdeep(1,1);   
    scanf("%d",&m);    
    int anss=0;
    int a=1,b;
    for (int i=0;i<m;i++)
         {
             scanf("%d",&b);
              anss+=deep[a]+deep[b]-2*deep[lca(a,b)];
              a=b;
                 }
    printf("%d",anss);
    return 0;                     
    }