这应该算一道树形背包吧,虽然我还是分不太清树形背包和树形dp的区别……
首先dp[i][u][j] 表示在u的前 i 棵子树中,留了 j 条树枝时最大的苹果数量,而且根据题目描述,这些留下的树枝一定都连在u上。
然后我们从1~j 枚举 k,于是
dp[i][u][j] = max(dp[i - 1][u][j], dp[i - 1][u][j - k] + dp[i - 1][v][k - 1] + cost[u->v])
思路就是我们让v所在的子树和u相连取更新答案。
j 的范围是枚举到当前的v时,u的子树中树枝的数量。
然后我们可以把第一维优化掉。
1 #include<cstdio> 2 #include<iostream> 3 #include<cmath> 4 #include<algorithm> 5 #include<cstring> 6 #include<cstdlib> 7 #include<cctype> 8 #include<vector> 9 #include<stack> 10 #include<queue> 11 using namespace std; 12 #define enter puts("") 13 #define space putchar(' ') 14 #define Mem(a, x) memset(a, x, sizeof(a)) 15 #define rg register 16 typedef long long ll; 17 typedef double db; 18 const int INF = 0x3f3f3f3f; 19 const db eps = 1e-8; 20 const int maxn = 105; 21 inline ll read() 22 { 23 ll ans = 0; 24 char ch = getchar(), last = ' '; 25 while(!isdigit(ch)) {last = ch; ch = getchar();} 26 while(isdigit(ch)) {ans = ans * 10 + ch - '0'; ch = getchar();} 27 if(last == '-') ans = -ans; 28 return ans; 29 } 30 inline void write(ll x) 31 { 32 if(x < 0) x = -x, putchar('-'); 33 if(x >= 10) write(x / 10); 34 putchar(x % 10 + '0'); 35 } 36 37 int n, q; 38 vector<int> v[maxn], c[maxn]; 39 40 bool vis[maxn]; 41 int edg[maxn], dp[maxn][maxn]; 42 void dfs(int now) 43 { 44 vis[now] = 1; 45 for(int i = 0; i < (int)v[now].size(); ++i) 46 { 47 if(!vis[v[now][i]]) 48 { 49 dfs(v[now][i]); 50 edg[now] += edg[v[now][i]] + 1; 51 for(int j = edg[now]; j > 0; --j) 52 for(int k = j; k > 0; --k) 53 dp[now][j] = max(dp[now][j], dp[now][j - k] + dp[v[now][i]][k - 1] + c[now][i]); 54 } 55 } 56 } 57 58 int main() 59 { 60 n = read(); q = read(); 61 for(int i = 1; i < n; ++i) 62 { 63 int x = read(), y = read(), co = read(); 64 v[x].push_back(y); c[x].push_back(co); 65 v[y].push_back(x); c[y].push_back(co); 66 } 67 dfs(1); 68 write(dp[1][q]); enter; 69 return 0; 70 }