#include<bits/stdc++.h>
using namespace std;
const int N=1e4+10;
vector<int>G[N];
int m,n,c[N],f[N][2];
/*
f[x][k]表示x结点染成k色使得x子树合法的最小染色数。
可以想到如果x的孩子y染了k色，那么y的k色完全可以染在x上，不需要染v,可以得到转移方程。 
f[x][0]+=min(f[y][0]-1,f[y][1]);
f[x][1]+=min(f[y][1]-1,f[y][0]);
*/
void dp(int x,int ff)
{
    if(x<=n) return;
    f[x][0]=f[x][1]=1;
    for(int y:G[x])if(y!=ff)
	{
        dp(y,x);
        f[x][0]+=min(f[y][0]-1,f[y][1]);
        f[x][1]+=min(f[y][1]-1,f[y][0]);
    }
}
int main()
{
    scanf("%d%d",&m,&n);
    memset(f,0,sizeof(f));
	for(int i=1;i<=n;i++){scanf("%d",&c[i]); f[i][c[i]]=1,f[i][!c[i]]=1e9;}
    for(int i=1,x,y;i<m;i++)
    {
        scanf("%d%d",&x,&y);
        G[x].emplace_back(y);
        G[y].emplace_back(x);
    }
    
	int rt=n+1;
    dp(rt,rt);
    
    printf("%d",min(f[rt][1],f[rt][0]));
    return 0;
}

这题真的有绿吗……

我看完题解之后有一个问题，为何哪一个节点作为根无所谓呢？

我们考虑两个节点$x$和$y$满足$x$是$y$的父亲，假设原树的子图$X$表示以$x$为根节点的子树除去以$y$为根节点的子树，设$y$的子树为$Y$。我们要证明哪一个节点作为根无所谓等价于证明$x$做父亲和$y$做父亲是一样的。

那为何他们是一样的呢？

不难想到如果$X$所需要的颜色与$Y$所需要的颜色一样，那么只需将父亲染色即可，那个节点作为父亲无所谓；如果所需颜色不一样，那么都需要各自的子树去染色，与父亲节点并无关联。

综上，哪一个作为根节点并无关系。