ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

叶子的染色(树型dp)

2019-08-19 11:42:59  阅读:279  来源: 互联网

标签:叶子 结点 int 染色 着色 树型 root dp


给一棵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

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

转移方程:dp[root][0]+=min(dp[e[i].to][0]-1,dp[e[i].to][1]);

dp[root][1]+=min(dp[e[i].to][1]-1,dp[e[i].to][0]);
代码:

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int INF=0x3f3f3f3f;
const int maxn=10005;
int m,n,c[maxn],head[maxn],cnt,dp[maxn][2];
struct Edge{
	int to,next;
}e[maxn*2];
void add_edge(int u,int v){
    e[cnt].to=v;
	e[cnt].next=head[u];
	head[u]=cnt++;
}
void dfs(int root,int fa){
    dp[root][0]=dp[root][1]=1;
    if(root<=n){
        if(c[root]==0)
            dp[root][1]=INF;
        else
            dp[root][0]=INF;
    }
    for(int i=head[root];i!=-1;i=e[i].next)
        if(e[i].to!=fa){
            dfs(e[i].to,root);
            dp[root][0]+=min(dp[e[i].to][0]-1,dp[e[i].to][1]);
            dp[root][1]+=min(dp[e[i].to][1]-1,dp[e[i].to][0]);
        }
}
int main(){
    memset(head,-1,sizeof(head));
    cin>>m>>n;
	cnt=0;
    for(int i=1;i<=n;i++) cin>>c[i];
    for(int i=1;i<m;i++){
    	int x,y;
        cin>>x>>y;
		add_edge(x,y);
		add_edge(y,x);
	}
    dfs(n+1,-1);
    cout<<min(dp[n+1][0],dp[n+1][1]);
    return 0;
}

标签:叶子,结点,int,染色,着色,树型,root,dp
来源: https://blog.csdn.net/qq_44238714/article/details/99717825

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有