洛谷 P2015 二叉苹果树 题解

题目

题目-二叉苹果树 - 洛谷

题目描述 

有一棵苹果树，如果树枝有分叉，一定是分二叉（就是说没有只有一个儿子的结点）

这棵树共有 𝑁 个结点（叶子点或者树枝分叉点），编号为 1∼𝑁，树根编号一定是 1。

我们用一根树枝两端连接的结点的编号来描述一根树枝的位置。下面是一颗有 4 个树枝的树：

2   5
 \ / 
  3   4
   \ /
    1

现在这颗树枝条太多了，需要剪枝。但是一些树枝上长有苹果。

给定需要保留的树枝数量，求出最多能留住多少苹果。

输入格式 

第一行 2 个整数 𝑁 和 𝑄，分别表示表示树的结点数，和要保留的树枝数量。

接下来 𝑁−1 行，每行 3 个整数，描述一根树枝的信息：前 2 个数是它连接的结点的编号，第 3 个数是这根树枝上苹果的数量。

输出格式

 一个数，最多能留住的苹果的数量。

代码 

#include<bits/stdc++.h>
using namespace std;
int f[105][3005],val[105][105],n,m;
vector<int> mp[3005];
void dp(int root,int fa){
	for(int i=0;i<mp[root].size();i++){
		int x=mp[root][i];
		if(x==fa) continue;
		dp(x,root);
		for(int j=m;j>=1;j--){
			for(int k=0;k<j;k++){
				f[root][j]=max(f[root][j],f[x][k]+f[root][j-k-1]+val[root][x]);
			}
		}
	}
}
int main(){
	cin>>n>>m;
	for(int i=1;i<n;i++){
		int u,v,x;
		cin>>u>>v>>x;
		val[u][v]=val[v][u]=x;
		mp[u].push_back(v);
		mp[v].push_back(u);
	}
	dp(1,0);
	cout<<f[1][m];
	return 0;
}