三色二叉树 ---伪树形dp
作者:互联网
题目描述
一棵二叉树可以按照如下规则表示成一个由0、1、2组成的字符序列,我们称之为“二叉树序列S”:
0 该树没有子节点 1S1 该树有一个子节点,S1为其二叉树序列 1S1S2 该树有两个子节点,S1,S2分别为两个二叉树的序列 例如,下图所表示的二叉树可以用二叉树序列S=21200110来表示。
你的任务是要对一棵二叉树的节点进行染色。每个节点可以被染成红色、绿色或蓝色。并且,一个节点与其子节点的颜色必须不同,如果该节点有两个子节点,那么这两个子节点的颜色也必须不相同。给定一棵二叉树的二叉树序列,请求出这棵树中最多和最少有多少个点能够被染成绿色。
题目大意:
给你一棵树,相邻点不同色,共三色,求一种颜色最多最少涂法。
分析:
根据题目条件列转移方程:
性质1.因为有三种颜色,所以如果父节点不是绿,那么两儿子必有一个是绿。(废话1)
性质2.如果父节点是绿,而两个子节点必然都不绿,且不冲突。(废话2)
我们不妨设置两种情况(新开一维度):dp[ rt ][ 0 ]表示 rt这个点不是绿色,dp[ rt ][ 1 ]表示是绿色,dp数组的值表示以这个点为根节点的树最多涂几个绿点。
我们可以给每一个节点设置初始值:dp[ rt ][ 0 ]=0,dp[ rt ][ 1 ]=1(很好理解吧。。。)
转移方程:1. dp[ rt ][ 1]=dp[ lch[ rt ] ] [ 0 ]+dp[ rch[ rt ] ] [ 0 ]+1;
该点涂绿情况:左儿子不涂绿情况+右儿子不涂绿情况+自己;
(如果这个点涂成绿色,那么左右两点都不可以涂成绿色,且这样的左右两点不冲突情况必定存在,即性质2)
2.dp[ rt ][ 0 ]=min/max ( dp[ lch[ rt ] ] [ 0 ]+dp[ rch[ rt ] ] [ 1 ] , dp[ lch[ rt ] ] [ 1]+dp[ rch[ rt ] ] [ 0 ] );
该点不涂绿情况:左儿子不涂绿,右儿子涂绿或者左儿子涂绿,右儿子不涂绿。
(也是必然条件,符合上面性质1)
然后就是dfs了
代码:
#include<cstdio> #include<iostream> #include<algorithm> #include<cstring> #include<queue> using namespace std; const int maxn=1e6+10; char s[maxn];int a[maxn]; int cnt=1;//用来确定dfs序,因为1是根了,所以已经有一个了(废话) int dfs[maxn]; int lch[maxn],rch[maxn],dp[maxn][2]; void build(int rt){//递归建树,自行理解,代码不唯一 if(a[rt]==1){ lch[rt]=++cnt; build(lch[rt]); }else if(a[rt]==2){ lch[rt]=++cnt; build(lch[rt]); rch[rt]=++cnt; build(rch[rt]); }else if(a[rt]==0){ return; } } void dfsmax(int rt){ dp[rt][1]=1;dp[rt][0]=0; if(lch[rt])dfsmax(lch[rt]); if(rch[rt])dfsmax(rch[rt]); dp[rt][1]+=dp[lch[rt]][0]+dp[rch[rt]][0]; dp[rt][0]=max(dp[lch[rt]][1]+dp[rch[rt]][0],dp[lch[rt]][0]+dp[rch[rt]][1]); //转移方程 } void dfsmin(int x){ dp[x][1]=1;dp[x][0]=0; if(lch[x]) dfsmin(lch[x]); if(rch[x]) dfsmin(rch[x]); dp[x][1]+=dp[lch[x]][0]+dp[rch[x]][0]; dp[x][0]=min(dp[lch[x]][1]+dp[rch[x]][0],dp[lch[x]][0]+dp[rch[x]][1]); } int main(){ scanf("%s",s); int len=strlen(s); for(int i=0;i<len;i++){ a[i+1]=s[i]-'0';//字符数组转整形,可忽略 } build(1); dfsmax(1);//以1为根节点建树并进行下列操作 printf("%d ",max(dp[1][1],dp[1][0]));//根节点里面保存的就是整个树的情况了 memset(dp,0,sizeof(dp));//记得初始化!!! dfsmin(1); printf("%d ",min(dp[1][1],dp[1][0])); return 0; }
标签:rt,int,dp,树形,rch,二叉树,节点,lch 来源: https://www.cnblogs.com/liu-yi-tong/p/12984192.html