商务旅行 code[vs] 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

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

分析

所以可以RMQ转LCA算法算出LCA。
距离的话可以定义个数组dis,dis[i]表示根节点到点i的距离。
点v,u的距离=dis[v]+dis[u]-2*dis[u,v的最近公共祖先]
dis可以用dfs预处理出。

代码(难看的c++)

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>

using namespace std;

struct arr{
    int x,y;
    int next;
};
arr edge[202001];
int n,m,nm;
int num;
int root;
int d[202001][21];
int f[202001][21];
int p[202001];
int a[202001],b[202001];
int ls[202001];
int v[202001];
int dis[202001];

int min(int x,int y)
{
    if (x>y)
      return y;
    else
      return x;
}

void add(int x,int y)
{
    nm++;
    edge[nm].x=x;
    edge[nm].y=y;
    edge[nm].next=ls[x];
    ls[x]=nm;
}

void init()
{
    memset(ls,0,sizeof(ls));
    memset(d,0,sizeof(d));
    memset(v,0,sizeof(v));
    memset(p,0,sizeof(p));
    memset(a,0,sizeof(a));
    memset(b,0,sizeof(b));

    scanf("%d",&n);
    for (int i=1;i<=n-1;i++)
      {
        int a,b;
        scanf("%d%d",&a,&b);
        add(a,b);
        add(b,a);
      }
}

void dp()
{   
    for (int i=1;i<=num;i++)
      {
        d[i][0]=a[i];
        f[i][0]=b[i];
       }

    for (int j=1;j<=20;j++)
      for (int i=1;i<=num;i++)
        {
          if (i+(1<<j)-1>num) break;
          d[i][j]=min(d[i][j-1],d[i+(1<<(j-1))][j-1]);
          if (d[i][j]==d[i][j-1])
            f[i][j]=f[i][j-1];
          else
            f[i][j]=f[i+(1<<(j-1))][j-1];
        }
}

int dfs(int r,int x,int dep)
{
    int i=ls[x];
    num++;
    p[x]=num;
    a[num]=dep;
    b[num]=x;
    while (i!=0)
      {
        if (edge[i].y==r) 
          {
            i=edge[i].next;
            continue;
          }
        dis[edge[i].y]=dis[x]+1;
        dfs(x,edge[i].y,dep+1);
        num++;
        a[num]=dep;
        b[num]=x;  
        i=edge[i].next; 
      }
}

int main()
{
    init();
    num=0;
    root=1;
    dfs(0,root,1);
    dp();
    scanf("%d",&m);
    int a,b;
    int ans2=0;
    scanf("%d",&b);
    ans2=dis[b];
    for (int i=2;i<=m;i++)
      {
        scanf("%d",&a);
        int a1=a;
        int b1=b;
        a=p[a];
        b=p[b];
        if (a>b) 
          {
            int c=a;
            a=b;
            b=c;
          }
        int ans;
        int k=floor(log(b-a+1)/log(2));
        ans=min(d[a][k],d[b-(1<<k)+1][k]);
        int shu;
        if (ans==d[a][k])
          shu=f[a][k];
        else
          shu=f[b-(1<<k)+1][k];
        ans2+=dis[a1]+dis[b1]-2*dis[shu];
        b=a1;
      }
    printf("%d\n",ans2);
    return 0;
}