bzoj 3910: 火车

最新推荐文章于 2018-11-01 19:52:00 发布

lqybzx

最新推荐文章于 2018-11-01 19:52:00 发布

阅读量1k

点赞数

分类专栏： LCA 文章标签： LCA

本文链接：https://blog.csdn.net/lqybzx/article/details/44872781

版权

LCA 专栏收录该内容

8 篇文章 0 订阅

订阅专栏

Description

A 国有n 个城市，城市之间有一些双向道路相连，并且城市两两之间有唯一

路径。现在有火车在城市 a，需要经过m 个城市。火车按照以下规则行驶：每次

行驶到还没有经过的城市中在 m 个城市中最靠前的。现在小 A 想知道火车经过

这m 个城市后所经过的道路数量。

Input

第一行三个整数 n、m、a，表示城市数量、需要经过的城市数量，火车开始

时所在位置。

接下来 n-1 行，每行两个整数 x和y，表示 x 和y之间有一条双向道路。

接下来一行 m 个整数，表示需要经过的城市。

Output

一行一个整数，表示火车经过的道路数量。

Sample Input

5 4 2
1 2
2 3
3 4
4 5
4 3 1 5

Sample Output

HINT

N<=500000 ,M<=400000

LCA+并查集。倍增LCA找两点间路径。并查集维护是否走过

#include<queue>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
using namespace std;
struct line
{
     int s,t;
     int next;
}a[1000001];
int head[500001];
int edge;
inline void add(int s,int t)
{
     a[edge].next=head[s];
     head[s]=edge;
     a[edge].s=s;
     a[edge].t=t;
}
bool v[500001];
int deep[500001];
int ans[500001][22];
inline void bfs(int r)
{
     int i,j;
     deep[r]=1;
     for(i=0;i<=21;i++)
          ans[r][i]=r;
     ans[r][0]=0;
     queue <int>Q;
     while(!Q.empty())
          Q.pop();
     Q.push(r);
     v[r]=true;
     while(!Q.empty())
     {
          int d=Q.front();
          Q.pop();
          for(i=head[d];i!=0;i=a[i].next)
          {
               int t=a[i].t;
               if(!v[t])
               {
                    v[t]=true;
                    Q.push(t);
                    deep[t]=deep[d]+1;
                    ans[t][0]=d;
                    int dt;
					long long dc;
                    for(j=1;j<=21;j++)
                    {
                    	 dt=ans[t][j-1];
                         ans[t][j]=ans[dt][j-1];
                    }
               }
          }
     }
}
long long ansx;
long long p[31];
inline int swim(int x,int y)
{
     while(deep[x]!=deep[y])
     {
          int i=0;
          while(deep[ans[y][i]]>deep[x])
               i++;
          if(i!=0)
               i--;
          ansx+=p[i];
          y=ans[y][i];
     }
     return y;
}
inline int lca(int x,int y)
{
	 ansx=0;
     if(deep[x]>deep[y])
     {
          int t=x;
          x=y;
          y=t;
     }
     y=swim(x,y);
     while(x!=y)
     {
          int i=0;
          while(ans[x][i]!=ans[y][i])
               i++;
          if(i!=0)
               i--;
          ansx+=p[i+1];
          x=ans[x][i];
          y=ans[y][i];
     }
     return y;
}
int fa[500001];
inline int find(int x)
{
     if(fa[x]!=x)
          fa[x]=find(fa[x]);
     return fa[x];
}
bool vis[500001];
int main()
{
     int n,m,st;
     scanf("%d%d%d",&n,&m,&st);
     int i;
     int s,t;
     for(i=1;i<=n-1;i++)
     {
          scanf("%d%d",&s,&t);
          edge++;
          add(s,t);
          edge++;
          add(t,s);
     }
     p[0]=1;
     for(i=1;i<=30;i++)
          p[i]=p[i-1]*(long long)2;
     bfs(1);
     for(i=1;i<=n;i++)
          fa[i]=i;
     long long as=0;
     int x;
     for(i=1;i<=m;i++)
     {
     	  scanf("%d",&x);
     	  if(fa[x]!=x)
     	       continue;
     	  int t=lca(st,x);
          as+=ansx;
          int fx=find(x),fy=find(st);
          while(fx!=fy)
          {
               if(deep[fx]<deep[fy])
                    swap(fx,fy);
               fa[fx]=find(ans[fx][0]);
               fx=fa[fx];
          }
          if(find(t)==t)
		       fa[t]=ans[t][0];
		  st=x;
     }
     printf("%lld\n",as);
     return 0;
}