树形DP

一、树形DP

问题：给一棵树，要求以最少的代价（或最大收益）完成给定的操作

二、步骤

• 1.确立状态：几乎所有的问题都要保存以某结点为根的子树的情况，再根据具体问题考虑是否要加维，加几维，如何加维。
• 2.状态转移：状态转移的变化比较多，要根据具体问题具体分析
• 3.实现方式:记忆化搜索（主要）或递推

三、例题

例1.由根分为左子树与右子树情况 二叉苹果树

• 大意：

first.本题权值在边上->将边权转移到儿子节点上

second.状态定义

• 设$f[i][j]$表示以i为根的子树上保留j个节点时最多能保留的苹果数量
• 设$ch[i][0],ch[i][1]分别为i的左右儿子$

third.状态转移：

• $f[i][j]=max(f[ch[i][0]][k]+f[ch[i][1]][j-k-1]+a[i](0<=k<=j-1)$
• $f[i][j]=0;(j=0时）$
• $f[i][j]=a[i](ch[i][0]=0且ch[i][1]=0,即叶子节点时)$
• $ans=f[1][m+1]$

CODE：记忆化搜索实现

#include<bits/stdc++.h>
using namespace std;
#define N 110
int n,m,a[N],f[N][N],ch[N][2],first[N],cnt;
struct node{
	int u,v,w,nxt;
}e[N*2];
inline void add(int u,int v,int w){
	e[++cnt].u=u;e[cnt].v=v;e[cnt].w=w;
	e[cnt].nxt=first[u];
	first[u]=cnt;
}
inline int read(){
    int data=0;int w=1; char ch=0;
    ch=getchar();
    while(ch!='-' && (ch<'0' || ch>'9')) ch=getchar();
    if(ch=='-') w=-1,ch=getchar();
    while(ch>='0' && ch<='9') data=(data<<3)+(data<<1)+ch-'0',ch=getchar();
    return data*w;
}
inline void dfs(int u,int f){
	for(int i=first[u];i;i=e[i].nxt){
		int v=e[i].v;
		if(v==f) continue;
		a[v]=e[i].w;
		if(ch[u][0]==0) ch[u][0]=v;
		else ch[u][1]=v;
		dfs(v,u);
	}
}
inline int getdp(int i,int j){
	if(i==0||j==0) return 0;
	if(~f[i][j]) return f[i][j];
	if(!ch[i][0] && !ch[i][1]) return a[i];
	for(int k=0;k<j;k++)
		f[i][j]=max(f[i][j],getdp(ch[i][0],k)+getdp(ch[i][1],j-k-1)+a[i]);
	return f[i][j];
}
int main(){
	n=read();
	m=read();
	memset(f,-1,sizeof(f));
	for(int i=1;i<n;i++){
		int u=read(),v=read(),w=read();
		add(u,v,w);
		add(v,u,w);
	}
	dfs(1,0);
	printf("%d",getdp(1,m+1));
	return 0;
}

例2.先把森林转化为二叉树，再dp 选课

• 题目中的依赖关系形成了一个森林，这是我们很不喜欢的

因此，我们，要将森林转化为二叉树

• 我们可以通过“左儿子右兄弟”来实现（其实还有另一种方法：通过不断加虚拟节点来实现）
• 1.将树中每个节点除了最左边的分支外全不删去。
• 2.连接每个节点以及它的兄弟
• 3.将树旋转45度即可
• 此时每个节点的左儿子是原来的儿子，右儿子就是它原来的兄弟
• 代码：

边读入边建立二叉树
for(int i=1;i<=n;i++){
		int k=read(),v=read();
		a[i]=v;
		if(k==0) k=n+1;
		if(son[k]==0) ch[k][0]=i;
		else ch[son[k]][1]=i;
		son[k]=i;
	}

状态定义：$f[i][j]$表示i的子树选j门课程所需的最大学分，则有：

转移方程：$f[i][j]=max(f[ch[i][1]][j],f[ch[i][0]][k]+f[ch[i][1]][j-k-1]+a[i])(0<=k<j)$

CODE

#include<bits/stdc++.h>
using namespace std;
inline int read(){
    int data=0;int w=1; char ch=0;
    ch=getchar();
    while(ch!='-' && (ch<'0' || ch>'9')) ch=getchar();
    if(ch=='-') w=-1,ch=getchar();
    while(ch>='0' && ch<='9') data=(data<<3)+(data<<1)+ch-'0',ch=getchar();
    return data*w;
}
const int N=310;
int n,m,f[N][N],a[N],ch[N][2],son[N];
inline int getdp(int i,int j){
	if(i==0||j==0) return 0;
	if(f[i][j]!=-1) return f[i][j];
	if(ch[i][1]) f[i][j]=getdp(ch[i][1],j);
	for(int k=0;k<j;k++)
		f[i][j]=max(f[i][j],getdp(ch[i][0],k)+getdp(ch[i][1],j-k-1)+a[i]);
	return f[i][j];
}
int main(){
	n=read();
	m=read();
	memset(f,-1,sizeof(f));
	for(int i=1;i<=n;i++){
		int k=read(),v=read();
		a[i]=v;
		if(k==0) k=n+1;
		if(son[k]==0) ch[k][0]=i;
		else ch[son[k]][1]=i;
		son[k]=i;
	}
	printf("%d",getdp(n+1,m+1));
	return 0;
}

• 复杂度：O(mn²)

例3 树的直径（最长链）

• 经过每个节点的最长链均为最长链+次长链
• 树形dp或两次dfs实现

CODE（树形dp方法）

#include<bits/stdc++.h>
using namespace std;
inline int read(){
    int data=0;int w=1; char ch=0;
    ch=getchar();
    while(ch!='-' && (ch<'0' || ch>'9')) ch=getchar();
    if(ch=='-') w=-1,ch=getchar();
    while(ch>='0' && ch<='9') data=(data<<3)+(data<<1)+ch-'0',ch=getchar();
    return data*w;
}
const int N=1e4+10;
struct node{
	int u,v,w,nxt;
}e[N*2];
int first[N],cnt,d1[N],d2[N];
inline void add(int u,int v,int w){
	e[++cnt].u=u;
	e[cnt].v=v;
	e[cnt].w=w;
	e[cnt].nxt=first[u];
	first[u]=cnt;
}
int n,a[N],son[N],ans;
inline void dfs(int u,int f){
	for(int i=first[u];i;i=e[i].nxt){
		int v=e[i].v;
		if(v==f) continue;
		dfs(v,u);
		if(e[i].w+d1[v]>d1[u]){
			d2[u]=d1[u];
			d1[u]=e[i].w+d1[v];
		}
		else if(e[i].w+d1[v]>d2[u]){
			d2[u]=e[i].w+d1[v];
		}
	} 
	ans=max(ans,d1[u]+d2[u]);
}
int main(){
	n=read();
	for(int i=1;i<n;i++){
		int u=read(),v=read(),w=read();
		add(u,v,w);
		add(v,u,w);
	}
	dfs(1,0);
	cout<<ans;
	return 0;
}

例4：树的中心：直径的中点

例5：战略游戏

• 题意：树上的最小点覆盖
• 法一：二分图匹配
• 法二：树形dp
• 直接上代码

CODE

#include<bits/stdc++.h>
using namespace std;
const int N=2e3+10;
struct node{
	int u,v,nxt;
}e[N*4];
int n,first[N],cnt,ch[N][2],f[N][2];
inline void add(int u,int v){
	e[++cnt].u=u;e[cnt].v=v;
	e[cnt].nxt=first[u];first[u]=cnt;
}
inline int read(){
    int data=0;int w=1; char ch=0;
    ch=getchar();
    while(ch!='-' && (ch<'0' || ch>'9')) ch=getchar();
    if(ch=='-') w=-1,ch=getchar();
    while(ch>='0' && ch<='9') data=(data<<3)+(data<<1)+ch-'0',ch=getchar();
    return data*w;
}

inline void dfs(int u,int fa){
	f[u][0]=0;f[u][1]=1;
	for(int i=first[u];i;i=e[i].nxt){
		int v=e[i].v;
		if(v==fa) continue;
		dfs(v,u);
		f[u][0]+=f[v][1];
		f[u][1]+=min(f[v][0],f[v][1]); 
	}
} 
int main(){
	n=read();
	memset(f,0x3f,sizeof(f));
	for(int i=1;i<=n;i++){
		int x=read(),k=read();
		for(int j=1;j<=k;j++){
			int t=read();
			add(x,t);add(t,x);
		}
	}
	dfs(0,-1);
	printf("%d",min(f[0][0],f[0][1]));
	return 0;
} 

完结！！！