HIHO #1055 : 刷油漆(树形dp 入门)

题目链接

入门的话还是看09《浅谈几类背包问题》 = =。

设dp[u][j] : 表示对于，以编号是u的节点为根的子树，选择不超过j个节点，可以获得的最大的价值。

状态的转移：dp[u][j]=max( dp[u][j] , dp[v_i][j-1] ) + value[u]（v_i是u的儿子，j>=1）

方程是类似完全背包的，儿子节点，使用dfs遍历一遍。

选择儿子节点的话，就必须选择父亲节点。假设存在u--v的一条边，

如果选择了v那么u是必须选择的，实现这一要求的是，我们在dfs访问v的时候，如何设置v的初始状态？

可以这样设置，首先v的状态等于他的父亲u的状态，dp[v][j=0...m-1]=dp[u][j=0...m-1],(m是最多能选择的节点的上限)，然后强制把节点v加入的状态里，dp[v][j=0...m-1] = dp[u][j=0...m-1] + value[v]

最后dfs回溯回来，也就是遍历完v的子孙节点后，更新v 的父亲。 dp[u][j] = max(dp[u][j], dp[v][j-1])(1<=j<=m)

代码实现加一个虚拟根节点0

答案就是dp[0][m];

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
#include<string>
#include<cmath>
#include<queue>
#include<map>
#include<set>
#include<cstdlib>
#include<vector>
using namespace std;
#define cl(a,b) memset(a,b,sizeof(a))
#define LL long long
#define pb push_back
#define gcd __gcd

#define For(i,j,k) for(int i=(j);i<k;i++)
#define lowbit(i) (i&(-i))
#define _(x) printf("%d\n",x)

const int maxn = 1e2+10;
const int inf  = 1 << 28;

int value[maxn];
vector<int> G[maxn];
int dp[maxn][maxn];

void dfs(int u,int M,int fa){
    if(M<=0)return ;
    for(int i=0;i<G[u].size();i++){
        int v = G[u][i];
        if(fa==v)continue;

        for(int j=0;j<M;j++){
            dp[v][j]=dp[u][j]+value[v];
        }

        dfs(v,M-1,u);

        for(int j=1;j<=M;j++){
            dp[u][j]=max(dp[u][j],dp[v][j-1]);
        }

    }
}

int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)scanf("%d",&value[i]);
    value[0]=0;
    int x,y;
    for(int i=0;i<n-1;i++){
        scanf("%d%d",&x,&y);
        G[x].pb(y);
        G[y].pb(x);
    }

    G[0].pb(1);
    dfs(0,m,-1);
    printf("%d\n",dp[0][m]);
    return 0;
}