1304: [CQOI2009]叶子的染色

1304: [CQOI2009]叶子的染色

原博主博客

Time Limit: 10 Sec Memory Limit: 162 MB

Description

给一棵m个结点的无根树，你可以选择一个度数大于1的结点作为根，然后给一些结点（根、内部结点和叶子均可）着以黑色或白色。你的着色方案应该保证根结点到每个叶子的简单路径上都至少包含一个有色结点（哪怕是这个叶子本身）。 对于每个叶结点u，定义c[u]为从根结点从U的简单路径上最后一个有色结点的颜色。给出每个c[u]的值，设计着色方案，使得着色结点的个数尽量少。

Input

第一行包含两个正整数m, n，其中n是叶子的个数，m是结点总数。结点编号为1，2，…，m，其中编号1，2，… ，n是叶子。以下n行每行一个0或1的整数（0表示黑色，1表示白色），依次为c[1]，c[2]，…，c[n]。以下m-1行每行两个整数a，b（1<=a < b <= m），表示结点a和b 有边相连。

Output

仅一个数，即着色结点数的最小值。

Sample Input

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

Sample Output

2

HINT

M<=10000
N<=5021

解析

• 找出隐藏信息：无根树，且任意一个非叶子节点作为根不影响答案；
• dp[i][k]表示以i为根的子树中，将i染成k色时子树最少的染色点；
• 由于要求到叶子节点最后有染色节点，可以将颜色先染在叶子节点上，从叶节点向根更新求解

CODE

#include<bits/stdc++.h>
using namespace std;
const int N=1e4+5;
const int INF=0x7f7f7f7f;
int read(){
	int s=0,f=1;
	char ch=getchar();
	while(!isdigit(ch)){
		if(ch=='-') f=-f;
		ch=getchar();
	}
	while(isdigit(ch)){
		s=(s<<3)+(s<<1)+ch-'0';
		ch=getchar();
	}
	return f*s;
}
struct fjy{
	int nxt,v;
}e[N<<1];
int fir[N],cnt;
void add(int u,int v){
	e[++cnt].v=v;
	e[cnt].nxt=fir[u];
	fir[u]=cnt;
}
int f[N][2],w[N];//f[i][k]表示以i为根的子树中，将i染成k色的子树中最少染色个数 
int n,m,u,v;
int min(int x,int y){
	return x<y?x:y;
}
void dfs(int u,int fa){//printf("u=%d;",u); 
	if(!f[u][0]&&!f[u][1])
	f[u][0]=f[u][1]=1;
	else return;
	for(int i=fir[u];i;i=e[i].nxt){
		if(e[i].v==fa) continue;
		dfs(e[i].v,u);
		if(f[e[i].v][1]!=INF&&f[e[i].v][0]!=INF)
		{
			f[u][0]+=min(f[e[i].v][0]-1,f[e[i].v][1]);
			f[u][1]+=min(f[e[i].v][1]-1,f[e[i].v][0]);
		}
		else
		f[u][!w[e[i].v]]++;//,printf("v=%d",e[i].v);
	}
}
int main(){
	m=read(),n=read();
	for(int i=1;i<=n;i++)
	w[i]=read();
	for(int i=1;i<m;i++){
		u=read(),v=read();
		add(u,v),add(v,u);
	}
	for(int i=1;i<=n;i++) f[i][w[i]]=1,f[i][!w[i]]=INF;
	dfs(n+1,0);
	printf("%d",min(f[n+1][1],f[n+1][0]));
	return 0;
}