1722: 叶子的染色

题目描述

给一棵有 mm 个节点的无根树，你可以选择一个度数大于 11 的节点作为根，然后给一些节点（根、内部节点、叶子均可）着以黑色或白色。你的着色方案应保证根节点到各叶子节点的简单路径上都包含一个有色节点，哪怕是叶子本身。

对于每个叶子节点 uu，定义 c_ucu 为从根节点到 uu 的简单路径上最后一个有色节点的颜色。给出每个 c_ucu 的值，设计着色方案使得着色节点的个数尽量少。

输入

第一行包括两个数 m,nm,n，依次表示节点总数和叶子个数，节点编号依次为 11 至 mm。

接下来 nn 行每行一个 00 或 11 的数，其中 00 表示黑色，11 表示白色，依次为 c_1,c_2,\cdots ,c_nc1,c2,⋯,cn 的值。

接下来 m-1m−1 行每行两个整数 a,ba,b，表示节点 aa 与 bb 有边相连。

输出

输出仅一个数，表示着色节点数的最小值。

样例输入

5 3
0
1
0
1 4
2 5
4 5
3 5

样例输出 

2

提示

数据范围与提示

数据	11	22	33	44	55	66	77	88	99	1010
mm	1010	5050	100100	200200	400400	10001000	40004000	80008000	1000010000	1000010000
nn	55	2323	5050	9898	197197	498498	20442044	40044004	50215021	49964996

#include<bits/stdc++.h>
using namespace std;
const int N=1e4+10,INF=1e8;
 
int read()
{
    int res=0,w=1;
    char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){res=(res<<3)+(res<<1)+(ch^48);ch=getchar();}
    return res*w;
}
void write(int x)
{
    if(x<0){putchar('-');x=-x;}
    if(x>9)write(x/10);
    putchar(x%10+'0');
}
 
int head[N],to[N<<1],nxt[N<<1],tot;
void add_edge(int x,int y)
{
    nxt[++tot]=head[x];
    head[x]=tot;
    to[tot]=y;
}
int f[N][3],n;
bool c[N];
void dfs(int u,int fa)
{
    if(u<=n)
    {
        f[u][c[u]]=1;
        f[u][(!c[u])]=INF;
        return ;
    }
    f[u][0]=f[u][1]=1;
    for(int i=head[u];i;i=nxt[i])
    {
        int v=to[i];
        if(v==fa)continue ;
        dfs(v,u);
        f[u][0]+=min(f[v][0]-1,f[v][1]);
        f[u][1]+=min(f[v][1]-1,f[v][0]);
    }
}
 
int main()
{
    int m;
    m=read(),n=read();
    for(int i=1;i<=n;i++)
        c[i]=read();
    for(int i=1;i<m;i++)
    {
        int x=read(),y=read();
        add_edge(x,y);add_edge(y,x);
    }   
    dfs(n+1,0);
    write(min(f[n+1][0],f[n+1][1]));
}