hiho#1055 : 刷油漆

hiho#1055 : 刷油漆

树形dp的基础题。
注意理解成 分成很多个背包。
一个可以涂m个点的背包，可以拆分成可以涂m1，m2，m3….的背包

#include<cstdio>
#include<algorithm>
#include<vector>
#include<cstring>
using namespace std;




const int N=300;


int n,m;
int v[N];
vector<int >ve[N];
int dp[N][N];//dp[i][j]以i为根节点，涂j个块
int vis[N];


void dfs(int now)
{
    vis[now]=1;
   // printf("size:%d\n",ve[now].size());
    for(int i=0;i<ve[now].size();i++)
    {
        int to=ve[now][i];


       // printf("now:%d to:%d\n",now,to);


        if(vis[to])
            continue;
        dfs(to);
        for(int j=m;j>=2;j--)
        {
            for(int k=1;k<j;k++)
            {
                dp[now][j]=max(dp[now][j],dp[now][j-k]+dp[to][k]);
            }
        }


    }
    return ;






}


int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&v[i]);
        }
        for(int i=0;i<=n;i++)
            ve[i].clear();


        for(int i=0;i<n-1;i++)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            ve[a].push_back(b);
            ve[b].push_back(a);
        }
        memset(dp,0,sizeof dp);
        memset(vis,0,sizeof vis);


        for(int i=1;i<=n;i++)
            dp[i][1]=v[i];




        dfs(1);




        printf("%d\n",dp[1][m]);


    }




    return 0;
}