SSL P1746 商务旅游

目录:

题目:

  商务旅游 题目

分析:

  题目跟我们说要求的是两点间的最短距离,我当然可以用dfs,但是,为了寻求更为高效的算法,我们采用了LCA算法
  但有人就会问了,LCA不是用来求两点的最近公共祖先吗?当然,你没有记错,但我们只需一个公式就一样可以推出最短距离: d e p [ i ] + d e p [ j ] 2 d e p [ L C A ( 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;
}

猜你喜欢

转载自blog.csdn.net/qq_35786326/article/details/80371314
SSL