bzoj 1304: [CQOI2009]叶子的染色

 1 #include<cstdio>
 2 #include<iostream>
 3 #define N 10009
 4 #define inf 100000
 5 using namespace std;
 6 int b[N],cnt,n,m,c[N],f[N][2],head[N],next[2*N],u[2*N];
 7 void jia(int a1,int a2)
 8 {
 9     cnt++;
10     next[cnt]=head[a1];
11     head[a1]=cnt;
12     u[cnt]=a2;
13     return;
14 }
15 void dp(int a1)
16 {
17     b[a1]=1;
18     if(a1<=m)
19       {
20         f[a1][c[a1]]=1;
21         f[a1][c[a1]^1]=inf;
22         return;
23       }
24     else
25       f[a1][0]=f[a1][1]=1;
26     for(int i=head[a1];i;i=next[i])
27       if(!b[u[i]])
28         {
29             dp(u[i]);
30             f[a1][0]+=min(f[u[i]][1],f[u[i]][0]-1);
31             f[a1][1]+=min(f[u[i]][1]-1,f[u[i]][0]);
32         }
33     return;
34 }
35 int main()
36 {
37     scanf("%d%d",&n,&m);
38     for(int i=1;i<=m;i++)
39       scanf("%d",&c[i]);
40     for(int i=1;i<n;i++)
41       {
42         int a1,a2;
43         scanf("%d%d",&a1,&a2);
44         jia(a1,a2);
45         jia(a2,a1);
46       }
47     dp(n);
48     printf("%d\n",min(f[n][0],f[n][1]));
49     return 0;
50 }