bzoj 1304（树形dp）

传送门
题解：定义dp[i][0/1]为当前子树在i点染成黑/白色的最少花费。可以这样定义的原因是当前子树的最优方案中，子树的根一定是染了色的。具体注释见代码。

#include<bits/stdc++.h>
using namespace std;
const int MAXN=1e4+2;
int head[MAXN],edge=0;
struct EDGE {
    int v,nxt;
}e[MAXN<<1];
int n,m,root;
int col[MAXN],dp[2][MAXN];
inline int read() {
    int x=0;char c=getchar();
    while (c<'0'||c>'9') c=getchar();
    while (c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
    return x;
}
inline void adde(int u,int v) {
    e[edge].nxt=head[u],e[edge].v=v,head[u]=edge++;
    e[edge].nxt=head[v],e[edge].v=u,head[v]=edge++; 
}
void dfs(int p,int fa) {
    dp[0][p]=dp[1][p]=1;
    if (p<=m) {
        if (col[p]==1) dp[0][p]=0x3f3f3f3f;
        else dp[1][p]=0x3f3f3f3f;
    }
    for (int i=head[p];~i;i=e[i].nxt) {
        int v=e[i].v;
        if (v^fa) {
            dfs(v,p);
            dp[0][p]+=min(dp[0][v]-1,dp[1][v]);//当前点染色，并且子树中少染一个点
            dp[1][p]+=min(dp[1][v]-1,dp[0][v]);
        }
    }
}
int main() {
//  freopen("bzoj 1304.in","r",stdin);
    memset(head,-1,sizeof(head));
    n=read(),m=read();
    for (int i=1;i<=m;++i) col[i]=read()+1;
    for (int i=1;i<n;++i) {
        int u=read(),v=read();
        adde(u,v);
    }
    dfs(m+1,0);
    printf("%d\n",min(dp[0][m+1],dp[1][m+1]));
    return 0;
}