Ural 1018

题意：一颗二叉苹果树树上结苹果，要求剪掉几棵枝，然后求保留Q根树枝能保留的最多到苹果数。

方法：典型的树形DP，第一次做，看不出来。。。。晓东有一种邻接表的常数优化方法以及免建树方法，但我还是觉得建树后思路更加清晰。

          还是老问题，寻找状态。这次的状态是ans[i][j]表示以i为树根，j条边能采摘的最大苹果数。与此同时，因为所有的苹果都结在边上，但是我们可以将之平移到叶子方向的节点上，这样根节点root就要假设有一条虚拟边接在其上，且其苹果数为0.这样不难获得状态转移方程

          ans[i][j] = max(ans[i.leftchild][k] + ans[i.rightchild][j-k-1]) +　i.val (i.leftchild表示其左孩子的编号，i.val表示i与其父节点相连的那条边上的苹果数)，这样对于第i个节点ans[i][j]就会必取i与其父节点相连的那条边最终形成j条边。当j=0时，ans[i][j] = 0。

          由于多了一条虚拟边，注意DP时边数要加一。

 

代码：

 

#include <iostream> #include <algorithm> #include <cstring> using namespace std; #define N 110 #define Q 110 struct TreeNode { int lc, rc; int val; } treeNode[N]; bool vis[N]; int apple[N][N]; int ans[N][Q]; int n, q; void Init() { memset(vis, 0, sizeof(vis)); memset(apple, -1, sizeof(apple)); memset(ans, -1, sizeof(ans)); memset(treeNode, 0, sizeof(treeNode));//顺便将tree[1].val 初始化为0 } void MakeTree(int root) { vis[root] = true; for (int i = 1; i <= n; i++) { if (!vis[i] && apple[root][i] != -1) { //vis[i] = true; if (treeNode[root].lc == 0) treeNode[root].lc = i; else treeNode[root].rc = i; treeNode[i].val = apple[root][i]; MakeTree(i); } } } int TreeDP(int root, int tq) { if (root == 0 || tq == 0) { return 0; } if (ans[root][tq] != -1) return ans[root][tq]; int max = 0; int lans, rans; for (int i = 0; i < tq; i++) { lans = TreeDP(treeNode[root].lc, i); rans = TreeDP(treeNode[root].rc, tq - i - 1); if (lans + rans > max) max = lans + rans; } ans[root][tq] = max + treeNode[root].val; return ans[root][tq]; } int main() { Init(); cin >> n >> q; int a, b, c; for (int i = 1; i <= n - 1; i++) { cin >> a >> b >> c; apple[a][b] = apple[b][a] = c; } MakeTree(1); cout << TreeDP(1, ++q) << endl; return 0; }