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;}