计算答案显然是一步一步的走然后求Lca用深度计算,关键就是怎么标记走过的路径
用并查集记录从这个点出发经过的最上端的城市,让后每次合并的时候从两个端点向lca合并,因为lca并查集记录的肯定是所有路径最上端的点
LCA再次写的树剖..
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<queue>
#include<vector>
#include<map>
#include<set>
#include<algorithm>
#include<iostream>
#define N 500500
#define ll long long
using namespace std;
int sc()
{
int i=0,f=1; char c=getchar();
while(c>'9'||c<'0'){if(c=='-')f=-1;c=getchar();}
while(c>='0'&&c<='9')i=i*10+c-'0',c=getchar();
return i*f;
}
int deep[N],top[N],fa[N],_fa[N],size[N];
int head[N],lst[N*2],nxt[N*2],a[N];
long long ans;
int n,m,st,tot;
void insert(int x,int y)
{
lst[++tot]=y;
nxt[tot]=head[x];
head[x]=tot;
}
void dfs(int x,int f)
{
size[x]=1;fa[x]=f;
for(int i=head[x];i;i=nxt[i])
if(lst[i]!=f)
{
deep[lst[i]]=deep[x]+1;
dfs(lst[i],x);
size[x]+=size[lst[i]];
}
}
void DFS(int x,int htp)
{
int k=0;top[x]=htp;
for(int i=head[x];i;i=nxt[i])
if(lst[i]!=fa[x]&&size[lst[i]]>size[k])k=lst[i];
if(!k)return;DFS(k,htp);
for(int i=head[x];i;i=nxt[i])
if(lst[i]!=fa[x]&&lst[i]!=k)
DFS(lst[i],lst[i]);
}
int Lca(int x,int y)
{
while(top[x]!=top[y])
{
if(deep[top[x]]<deep[top[y]])swap(x,y);
x=fa[top[x]];
}
return deep[x]<deep[y]?x:y;
}
int find(int x)
{
return _fa[x]==x?x:_fa[x]=find(_fa[x]);
}
int main()
{
n=sc(),m=sc(),st=sc();
for(int i=1;i<n;i++)
{
int x=sc(),y=sc();
insert(x,y);
insert(y,x);
}
for(int i=1;i<=m;i++)a[i]=sc();
dfs(1,0);DFS(1,1);
for(int i=1;i<=n;i++)_fa[i]=i;
for(int i=1;i<=m;i++)
{
int fx=find(st),fy=find(a[i]);
if(fx==fy)continue;
int f=Lca(st,a[i]);
ans+=deep[st]+deep[a[i]]-2*deep[f];
int x=st,y;
while(find(x)!=find(f))
{
y=find(x);
_fa[y]=find(f);
x=fa[y];
}
x=a[i],y;
while(find(x)!=find(f))
{
y=find(x);
_fa[y]=find(f);
x=fa[y];
}
st=a[i];
}
cout<<ans;
return 0;
}