rt,后来用另一种方法过了,但有没有大佬能帮忙看看这个做法为什么会 TLE 两个点?
#include<bits/stdc++.h>
using namespace std;
int n,m,x,y,a[1000005],fa[1000005],d[1000005];
int head[1000005],ne[1000005],ver[1000005],tot;
void add(int x,int y){
ver[++tot]=y,ne[tot]=head[x],head[x]=tot;
}
int sum;
void dfs(int x,int f,int k){
fa[x]=f,d[x]=k;
for(int i=head[x];~i;i=ne[i]) if(ver[i]!=f) dfs(ver[i],x,k+1);
}
bool flag[1000005];
void merge(int x,int y){
flag[x]=1,flag[y]=1;
while(x!=y){
if(d[x]<d[y]) swap(x,y);
x=fa[x],flag[x]=1;
}
}
int main(){
cin>>n>>m;
memset(head,-1,sizeof(head));
for(int i=1;i<n;i++){
cin>>x>>y;
add(x,y),add(y,x);
}
for(int i=1;i<=m;i++) cin>>a[i];
dfs(1,0,1);
if(m==1){
cout<<1;
return 0;
}
for(int i=1;i<m;i++) merge(a[i],a[i+1]);
for(int i=1;i<=n;i++) sum+=flag[i];
cout<<sum;
return 0;
}