ZOJ 3201 Tree of Tree(树形DP+背包)

题目链接：点击打开链接

题意：给一棵节点带权树，找到一个有k个节点的子树，这个子树的有最大权值

思路：树形dp+背包。

dp[i][j]表示以i为根节点且有j个节点的子树的最大权值，对i的每个子节点做分组背包，选择适合数量的节点分配给它。

dp[i][j] = max{ max{dp[i][j-t] + dp[v][t] | 1<=t<j} | v是i的儿子节点}，ans = max{dp[i][k] | 0<=i<n}

// ZOJ 3201 Tree of Tree.cpp 运行/限制：0ms/1000ms
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <iostream>
using namespace std;
int n, k, re;
int dp[105][105];
int weight[105], cnt[105];
vector<int> tree[105];
void init() {
	re = 0;
	memset(dp, 0, sizeof(dp));
	for (int i = 0; i < 105; i++) {
		tree[i].clear();
	}
}
int dfs(int node, int fa) {
	cnt[node] = 1;
	for (int i = 0; i < tree[node].size(); i++) {
		int v = tree[node][i];
		if (v == fa) {
			continue;
		}
		cnt[node] += dfs(v, node);
	}
	dp[node][1] = weight[node];
	for (int i = 0; i < tree[node].size(); i++) {
		int v = tree[node][i];
		for (int j = cnt[node]; j >= 1; j--) {//此处顺序不可乱，因为要求dp[node][j-t]是以之前的子节点为根的子树产生的;
			                                  //dp[v][t]是以当前子节点为根的子树产生的;
			                                  //如果循环顺序反过来，dp[node][j-t]就可以是以当前子节点为根的子树产生的;
			for (int t = 0; t < j && t <= cnt[v]; t++) {
				dp[node][j] = max(dp[node][j], dp[node][j - t] + dp[v][t]);//dp[node][j-t]是以之前的子节点为根的子树产生的;
				                                                           //dp[v][t]是以当前子节点为根的子树产生的;
			}
		}
	}
	re = max(re, dp[node][k]);
	return cnt[node];
}
int main(){
	int a, b;
	while (scanf("%d%d", &n, &k) != EOF) {
		init();
		for (int i = 0; i < n; i++) {
			scanf("%d", &weight[i]);
		}
		for (int i = 1; i < n; i++) {
			scanf("%d%d", &a, &b);
			tree[a].push_back(b);
			tree[b].push_back(a);
		}
		dfs(0, -1);
		printf("%d\n", re);
	}
    return 0;
}