记录编号 611438 评测结果 AAAAAAAAAAAA
题目名称 [THUPC 2025 pre] 辞甲猾扎 最终得分 100
用户昵称 Gravatar梦那边的原神 是否通过 通过
代码语言 C++ 运行时间 7.569 s
提交时间 2026-01-30 18:38:04 内存使用 84.67 MiB
显示代码纯文本
#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;
const int N=1e6+10;
int n,k,mk[N],vis[N];
int dp[N][4];
vector<int>G[N];
// dp[x][0] 不染色 
// dp[x][1] 染白色
// dp[x][2] 不染色,父亲有一个染白色
// dp[x][3] 不染色,儿子有一个染白色 
void add(int x,int y){
	G[x].push_back(y);
}
void dfs(int x,int fa){
	int val=N;
	dp[x][1]=1;
	for(auto y:G[x]){
		if(y==fa)continue;
		dfs(y,x);
		dp[x][0]+=min(min(dp[y][0],dp[y][1]),dp[y][3]);
		dp[x][1]+=min(min(dp[y][0],dp[y][1]),min(dp[y][2],dp[y][3]));
		dp[x][2]+=min(min(dp[y][0],dp[y][1]),dp[y][3]);
		dp[x][3]+=min(min(dp[y][0],dp[y][1]),dp[y][3]); 
		val=min(val,dp[y][1]-min(min(dp[y][0],dp[y][1]),min(dp[y][2],dp[y][3])));                                     
	}
	dp[x][3]+=val;
	if(mk[x])dp[x][1]+=N;
	if(vis[x])dp[x][0]+=N; 
	return;
}
int main(){
	freopen("thupc_2025_pre_J.in","r",stdin);
	freopen("thupc_2025_pre_J.out","w",stdout);
	scanf("%d %d",&n,&k);
	for(int i=1,x;i<=k;i++){
		scanf("%d",&x);
		mk[x]=1;
	}
	for(int i=1,u,v;i<n;i++){
		scanf("%d %d",&u,&v);
		add(u,v),add(v,u);
	}
	for(int i=1;i<=n;i++){
		if(!mk[i])for(auto j:G[i])vis[i]|=mk[j];
	}
	dfs(1,0);
	printf("%d\n",min(min(dp[1][0],dp[1][1]),dp[1][3]));
	return 0;
}