树形DP

版权声明:本文为博主原创文章,遵循 CC 4.0 by-sa 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/forever_dreams/article/details/81030372

今天上课的内容就是树形DP,新来的ldw老师第一次给我们上课,激动~

 

【例题引入】

题目传送门二叉苹果树

题目大意:有一颗二叉树(以1为根节点),每条边上有一些苹果,其中有 N 个节点,(N - 1)条边,现要减去一些边,保留 M 条边,使得剩下的苹果个数最多,求出这个数(洛谷 P2015)

其实我看到题如果不会树形DP的话连暴力都不会打的

这道题很像一道DP题(最优解),但普通的DP肯定是不行的,我们要用树形DP来做

 

【基本思想】

所谓树形DP呢,就是在树上做DP

DP方向有两个:

  1. 根--->叶:即从根节点转移到叶节点,这种动态规划在实际的问题中运用的不多 
  2. 叶--->根:即根的子节点传递有用的信息给根,然后根得出最优解的过程,这类的题目比较多

树是一种特殊的图,可以描述比较复杂的关系,再加上树的递归定义,是一种非常合适动规的框架,树形动态规划就成为动规中很特殊的一种类型。

我们知道,动态规划类问题必须满足两个性质——最优子结构性质无后效性,现在将DP转移到树上,(以例题举例)该问题具有明显的最优子结构性质;每个问题都与左右儿子结点有关系,但不与孙子结点发生关系,具备无后效性;且计算方案时,搜索子结构时具备重叠性,可以用动态规划来解决。 

那么和其他的动态规划题目一样,树形DP的解决也要考虑3步:

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

其实只要把状态转移方程写出来一般就问题不大了

 

【例题分析】

本题的权值在边上,不太好用这个信息来解决问题,我们先把问题转换一下,选 M 条边转换成选(M + 1)个点,且必须选择根节点(不然整棵树就没了)

对于一个节点 x,它的左儿子为 lson [ x ]右儿子为 rson [ x ]

定义 f [ x ][ y ] 表示以 x 为根的子树上保留 y 个节点时的最大苹果数,那答案就是 f [ 1 ][ M + 1 ]

现在我们考虑如何转移

我们假设现在 dp 到以 x 为根节点的子树,它能保留 y 个点,若我们给左儿子 i 个点,那右儿子就只有(y - i - 1)个点(根节点必须要保留),我们只需要枚举 i 就可以了

状态转移方程:f [ x ][ y ] = max { f [ lson [ x ] ] [ i ] + f [ rson [ x ] ][ y - i - 1 ] + a [ i ] } ( 0 ≤ i ≤ y - 1 ) 

初始化:这个DP是从叶节点做到根节点的,因此叶节点初始化为它与它父亲连的边的苹果数,其他节点的初始化为0

这道题中是知道根节点为1,若不知道的话还要枚举根节点,再建树

 

【代码】

好的,上代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=105,M=205;
int n,m,t;
int first[N],v[M],w[M],next[M];
int a[N],lson[N],rson[N],father[N],f[N][N];
bool used[N];
void add(int x,int y,int z)
{
    t++;
    next[t]=first[x];
    first[x]=t;
    v[t]=y;
    w[t]=z;
}
int dp(int x,int y)
{
    if(x==0||y==0)  return 0;
    if(!lson[x]&&!rson[x])  return a[x];
    if(f[x][y]>0)  return f[x][y];
    int i,j;
    for(i=0;i<=y-1;++i)
      f[x][y]=max(f[x][y],dp(lson[x],i)+dp(rson[x],y-i-1)+a[x]);
    return f[x][y];
}
void pre(int root)
{
    int x,i;
    for(i=first[root];i;i=next[i])
    {
        x=v[i];
        if(x==father[root])
          continue;
        father[x]=root;
        if(!lson[root])  lson[root]=x;
        else  rson[root]=x;
        a[x]=w[i];
        if(!used[x])
        {
            used[x]=true;
            pre(x);
        }
    }
}
int main()
{
    int x,y,z,i;
    scanf("%d%d",&n,&m);
    for(i=1;i<n;++i)
    {
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z);
        add(y,x,z);
    }
    pre(1);
    dp(1,m+1);
    printf("%d",f[1][m+1]);
    return 0;
}

 

【一些小知识】

如果题目中要给出一棵树,除了直接说给出一棵树之外,还有以下几种常见说法:

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

展开阅读全文

没有更多推荐了,返回首页