F - Vlad and Unfinished Business（树上操作，暴力跳根）

F - Vlad and Unfinished Business

记录每个需要走过的点往根节点，即初始点x跳，每次跳的时候+2，因为要过去再回来，如果已经父节点已经跳过，那就直接找下一个点。

#include<bits/stdc++.h>
#include<stdio.h>
typedef long long ll;
using namespace std;
const int N=1e6+10,mod=998244353;
vector<int>vec[N],seq;
ll ans;
bool has[N];
int x,y,ne[N];
void dfs(int u,int fa,int step)
{
	ne[u]=fa;//记录u这个节点的父节点
	if(u==y){
		ans=-step;//如果搜到了y，那么就要减掉y的深度，因为到了y就不用再往后走到x了。
                //然后这里也要继续往下搜，因为一些节点的父节点可能是y，
                //防止他的子节点的父节点变成了0
	}
	for(int i=0;i<vec[u].size();i++)
	{
		int t=vec[u][i];
		if(t==fa)continue ;
		dfs(t,u,step+1);
	}
}
int main()
{
	int t;
	cin>>t;
	while(t--)
	{
		seq.clear();
		ans=0;
		int n,k;
		cin>>n>>k;
		for(int i=1;i<=n;i++)vec[i].clear(),has[i]=false;//初始化
		cin>>x>>y;
		while(k--)
		{
			int tt;
			cin>>tt;
			seq.push_back(tt);//存要经过的点
		}
		seq.push_back(y);//终点也要存
		for(int i=0;i<n-1;i++)
		{
			int l,r;
			cin>>l>>r;
			vec[l].push_back(r);//存双向边
			vec[r].push_back(l);
		}
		dfs(x,-1,0);
//		cout<<ans<<'\n';
		has[x]=true;
		for(int i=0;i<seq.size();i++)//求要经过的点，往根节点跳
		{
			int t=seq[i];
			for(int u=t;!has[u];u=ne[u])//如果u这个点还没被走过，那就走，否则结束
			{
				ans=ans+2;//每次+2
				has[u]=true;
			}
		}
		cout<<ans<<'\n';
	}
}