| 记录编号 |
611438 |
评测结果 |
AAAAAAAAAAAA |
| 题目名称 |
[THUPC 2025 pre] 辞甲猾扎 |
最终得分 |
100 |
| 用户昵称 |
梦那边的原神 |
是否通过 |
通过 |
| 代码语言 |
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;
}