【hiho一下第十二周】刷油漆

【题目链接】:http://hihocoder.com/problemset/problem/1055

【题意】

【题解】

设f[x][i]表示以第x个节点为根的子树;
不选x这个节点,然后子树里面选i个其他点能够获得的最大价值;
在枚举儿子y的时候进行DP,第一层枚举这个子树里面选了多少个节点i->逆序枚举,用01背包的更新方式节省空间;
第二层,枚举这个儿子的子树里面选了几个节点k(也不包括这个儿子节点);
则
f[x][i] = max(f[x][i],f[x][i-k]+f[y][k-1]+v[y])
//因为f[x][i]不包括x节点本身,然后y以下有k个节点,则对应的状态应该是f[y][k-1]+v[y];
在根节点(任选)上面再加1个0号节点;
最后直接输出f[0][m]

【Number Of WA】

0

【完整代码】

#include <bits/stdc++.h>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define LL long long
#define rep1(i,a,b) for (int i = a;i <= b;i++)
#define rep2(i,a,b) for (int i = a;i >= b;i--)
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define ms(x,y) memset(x,y,sizeof x)

typedef pair<int,int> pii;
typedef pair<LL,LL> pll;

const int dx[9] = {0,1,-1,0,0,-1,-1,1,1};
const int dy[9] = {0,0,0,-1,1,-1,1,-1,1};
const double pi = acos(-1.0);
const int N = 110;

int n,m,v[N],f[N][N];
vector <int> g[N];

void dfs(int x,int num,int fa)
{
    for (int y:g[x])
    {
        if (y==fa) continue;
        dfs(y,num-1,x);
        rep2(i,num,0)
            rep1(k,1,i)
                f[x][i] = max(f[x][i],f[x][i-k]+f[y][k-1]+v[y]);
    }
}

int main()
{
    //freopen("F:\\rush.txt","r",stdin);
    ios::sync_with_stdio(false),cin.tie(0);//scanf,puts,printf not use
    cin >> n >> m;
    rep1(i,1,n)
        cin >> v[i];
    rep1(i,1,n-1)
    {
        int x,y;
        cin >> x >> y;
        g[x].pb(y),g[y].pb(x);
    }
    g[0].pb(1);
    dfs(0,m,-1);
    cout << f[0][m] << endl;
    return 0;
}