动态规划 —— 树形 DP

【概述】

树形动态规划是在树的数据结构上的动态规划,在各个阶段呈现树状关系的时候可以采用树形 DP,其基本思想是由子节点的信息推出父节点的信息。

树形 DP 中,是通过以下 4 点树的特点来进行建图的

  1.  n 个点,n-1 条边的无向图,任意两顶点间可达
  2. 无向图中任意两个点间有且只有一条路
  3. 一个点至多有一个前趋,但可以有多个后继
  4. 无向图中没有环

【计算顺序】

计算顺序与线性动态规划的顺推、逆推相似,同样有两个方向: 

  1. 叶 => 根:在回溯时,从叶节点向上转移状态
  2. 根 => 叶:在叶向根 DFS 一遍后(预处理),再重新向下获取答案

【表示方法】

树形 DP 还涉及到建图的问题,如果题目能很清晰的输入一个树,一般使用 vector s[N],用 s[i] 来保存节点 i 的所有儿子。

当树是一般树且涉及到权值时,一般采用链式存储法,即用结构体数组 edge 存边,edge[i] 表示第 i 条边,head[i] 存以 i 为起点的第一条边(在edge中的下标)

struct Node{
    int next; //下一条边的存储下标
    int to; //这条边的终点
    int w; //权值
}edge[N*2]; //由于是无向图,因此要开2倍
 

常用的建图方法如下

for(int i=1;i<=n;i++){
    int x,y;
    scanf("%d%d",&x,&y);
    addEdge(x,i);
    w[i]=y;
}

若以点 i 为起点的边新增了一条,则在 edge 中的下标为 j,那么 edge[j].next=head[i],然后head[i]=j,即每次新加的边作为第一条边,最后倒序遍历即可。

int cnt;//边的计数
void Add(int x,int y,int w) {    //起点x, 终点y, 权值w
    edge[cnt].x=x;
    edge[cnt].w=w;
    edge[cnt].next=head[x];
    head[y]=cnt++;
}

遍历以 x 为起点的边时,以 i 开始为第一条边,每次指向下一条(以-1为结束标志)

memset(head,-1,sizeof(head));
for(int i=head[st]; i!=-1; i=edge[i].next){
    ...
}

【计算方法】

树形 DP 的计算方法则与线性 DP 不同,线性 DP 一般采用传统迭代,而树是通过递归定义的,因此树形 DP 要递归来求解。

一般而言,树形 DP 常与背包问题结合起来,常用 dp[node] 表示的是以 node 为根的子树能得到的最优解,dp[node] 需要从 node 的子结点进行状态转移,由于 dp[node] 的状态是由它的儿子转移而来,因此可以 node 的 n 个儿子看做 n 个物品,对这 n 个物品抉择得到最优的 dp[node] 就用到背包的思想。

当然,树形 DP 不止 node 这一维,第二维一般都是以题目给出的限制需要来,然后根据题意保存好状态,写出状态转移方程,递归的求解 DP[root]

【基本步骤】

1.若问题是一棵隐性树(不以树为直接背景,但各个阶段呈树状关系),则需要将问题转化为一棵显性树,并存储各阶段的树状联系。

2.根据题目要求与数据量,选择合适的树的存储方式。

若节点数小于5000,一般选用邻接矩阵存储;若节点数大于5000,一般选用邻接表来存储(边要开到2*N,因为是无向图);若是二叉树或需要多叉转二叉,则可以用两个一维数组 brother[N]、child[N] 来存储。

3.写出动归方程

通过孩子和父亲之间的关系建立方程,根据要求选用 根 => 叶 或 叶 => 根 的计算方式。

【题目类型】

1.常规问题

  1. 没有上司的舞会(洛谷-P1352)(经典题,入门必做):点击这里
    英文版:Anniversary party(HDU-1520):点击这里
  2. Computer(HDU-2196)(经典题,入门必做):点击这里

2.树形背包问题

求在树上选一些点满足价值最大的问题,一般可设 dp[i][j] 表示 i 这颗子树选 j 个点的最优解。

  1. 有线电视网(洛谷-P1273)(分组背包):点击这里
    英文版:TELE(POJ-1115):点击这里
  2. Starship Troopers(HDU-1011)(分组背包,注意特判m=0的情况):点击这里
  3. Rebuilding Roads(POJ-1947)(分组背包):点击这里
  4. The more, The Better(HDU-1561)(01背包+虚拟根节点):点击这里

3.删点或删边问题

统计子树和,通过加减一些子树满足题目中要求的某些性质。

  1. Garland(CF-767C)(点权固定,注意特判):点击这里
  2. 最大子树和(洛谷-P1122)(根据具体权值大小来判断):点击这里
  3. 重建道路(洛谷-P1272)(删边+01背包):点击这里

你可能感兴趣的:(动态规划——树形,DP,——动态规划——)