J. 辞甲猾扎

#include <bits/stdc++.h>
using namespace std;
vector<int>a[1000005];
int id[1000005];
bool c[1000005],b[1000005];
long long f[1000005][3];
/*
对于与黑色点相邻的灰色点： 
0:根节点在之后被父节点染成白色
1:根节点在之后被子节点染成白色
2:根节点初始时染成白色
为了尽量简化问题，尝试把另外两种情况统一到这种DP状态中，为此使用更通用的描述，使用"覆盖"的定义：
0:根节点未能被覆盖 
1:根节点已经被覆盖 
2:根节点染成白色 
*/
void dfs(int n1,int fa)
{for(int i=0;i<a[n1].size();i++){if(a[n1][i]!=fa){dfs(a[n1][i],n1);}}if(b[n1]==false){f[n1][2]=1;for(int i=0;i<a[n1].size();i++){if(a[n1][i]!=fa){f[n1][2]+=min(f[a[n1][i]][0],min(f[a[n1][i]][1],f[a[n1][i]][2]));}}}else{f[n1][2]=INT_MAX;}f[n1][0]=0;for(int i=0;i<a[n1].size();i++){if(a[n1][i]!=fa){f[n1][0]+=min(f[a[n1][i]][1],f[a[n1][i]][2]);}}long long minn=INT_MAX;for(int i=0;i<a[n1].size();i++){if(a[n1][i]!=fa){minn=min(minn,f[a[n1][i]][2]-f[a[n1][i]][1]);}}if(minn<0){minn=0;}f[n1][1]=f[n1][0]+minn;if(c[n1]==false){f[n1][1]=min(f[n1][0],f[n1][1]);}
}
int main()
{ios::sync_with_stdio(false);cin.tie(0);int n,k;cin>>n>>k;for(int i=1;i<=k;i++){cin>>id[i];b[id[i]]=true;}for(int i=1;i<n;i++){int u,v;cin>>u>>v;a[u].push_back(v);a[v].push_back(u);}for(int i=1;i<=k;i++){for(int j=0;j<a[id[i]].size();j++){c[a[id[i]][j]]=true;}}for(int i=1;i<=k;i++){c[id[i]]=false;}dfs(1,0);cout<<min(f[1][1],f[1][2])<<endl;return 0;
}