The more, The Better
树形背包。
代码:
1 /* 2 hdu 1561 : The more,the better 3 4 dp[i][j] : 当前i节点及其子树下选择j个城市的最大值为dp[i][j]; 5 6 这一道题目注意建立了一个超级根节点0, 7 任何点都要先拿0才可以。所以后面很多地方都是m+1 8 9 下面的转移方程中有一句dp[u][j] = max(dp[u][j],dp[u][j-k]+dp[v][k]);10 那么可能就有疑问了:dp[u][j-k]+dp[v][k] 11 从子树v中取出k个点 与 根树u中取出的j-k个点 合并成 从根树中取j个点12 仔细读完就会发现问题了:既然v是u的子树,那么dp[v][k]会不会和dp[u][j-k]冲突呢13 即:从子树v中取出K个点,然后在u中取出j-k个点,会不会v中的点又在u个点中出现了呢14 答案是不会的。15 因为遍历每个从u遍历v,递归求解出dp[v]的值,用v来更新u,也就是说在以前是没有便利到v16 的,所以dp[u]中也不是由v更新的,所以dp[u][j-k]的点没有子树v中的点 17 18 然后注意转移时m一定要逆序,和01背包相似19 01:背包加入一个物品,然后用这个物品更新20 f[v],f[v]=max(f[v],f[v-Tiji[i]]+jiazhi[i])21 树上的相似:加入一个物品,即遍历到v,然后用这个物品更新22 dp[v] = max(dp[v],dp[v-k]+(新节点)子树大小为k的最大价值) //后面的那个递归求解。 23 然后由于树形背包的限制(拿v之前先拿u)加了一维。 24 25 26 */27 28 #include29 #include 30 #include 31 32 using namespace std;33 34 const int MAXN = 210;35 const int MAXM = 50010;36 37 struct Edge{38 int to,nxt;39 }e[MAXM];40 int head[MAXM],tot;41 int val[MAXN],dp[MAXN][MAXN];42 43 inline int read() {44 int x = 0,f = 1;char ch = getchar();45 for (; ch<'0'||ch>'9'; ch = getchar())46 if (ch=='-') f = -1;47 for (; ch>='0'&&ch<='9'; ch = getchar())48 x = x*10+ch-'0';49 return x*f;50 }51 inline void add_edge(int u,int v) {52 e[++tot].to = v,e[tot].nxt = head[u],head[u] = tot;53 }54 inline void init() {55 memset(head,0,sizeof(head));56 memset(dp,0,sizeof(dp));57 tot = 0;58 }59 void dfs(int u,int m) {60 dp[u][1] = val[u];61 for (int i=head[u]; i; i=e[i].nxt) {62 int v = e[i].to;63 dfs(v,m);64 for (int j=m; j>=2; --j) 65 for (int k=0; k