题目描述
有一棵二叉苹果树,如果数字有分叉,一定是分两叉,即没有只有一个儿子的节点。这棵树共 个节点,标号 至 ,树根编号一定为 。
我们用一根树枝两端连接的节点编号描述一根树枝的位置。一棵有四根树枝的苹果树,因为树枝太多了,需要剪枝。但是一些树枝上长有苹果,给定需要保留的树枝数量,求最多能留住多少苹果。
输入格式
第1行2个数,N和Q(1<=Q<= N,1<N<=100)。
N表示树的结点数,Q表示要保留的树枝数量。接下来N-1行描述树枝的信息。
每行3个整数,前两个是它连接的结点的编号。第3个数是这根树枝上苹果的数量。
每根树枝上的苹果不超过30000个。
输出格式
输出仅一行,表示最多能留住的苹果的数量。
样例
样例输入
5 2
1 3 1
1 4 10
2 3 20
3 5 20
样例输出
21
题目分析
可以很容易的看出,这是一道树形dp的题目。
但仅仅树形dp还是不够的,因为这道题限制了要保留的树枝的数量。
- 状态表示:
f[u][j] //表示以u节点为根的树,保留j个节点后可以保留的最大苹果数
- 状态计算:
状态f[u][j],可以由f[u][j-k]+f[v][k]+a[u][v]转移过来。
因此状态转移方程为:f[u][j]=max(f[u][j],f[u][j-k]+f[v][k]+a[u][v]);
- 注意:
1)f[][]数组一开始要初始化为负无穷。这么初始化是为了保证子节点只能由其父节点转移而来。
2)最后的结果为f[1][q+1]。因为f[u][j]表示的是保留了j个节点,而题目要求是保留q条边。而 树的边数=节点数+1。
代码如下
#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <map>
#include <queue>
#include <vector>
#include <set>
#include <algorithm>
#include <iomanip>
#define LL long long
#define PII pair<int,int>
using namespace std;
const int N=100+5;
vector<int> h[N];
int a[N][N]; //我们可以用邻接矩阵的方式来树a[u][v]=u-v这条树枝上的苹果数
int f[N][N],s[N]; //s[u]表示以u为根节点的树的节点数
void dfs(int u,int fa)
{
s[u]=1;
f[u][1]=0; //只保留1个节点的情况下,没有边存在,因此f[u][1]=0
for(int i=0;i<h[u].size();i++)
{
int v=h[u][i]; //dfs的基本操作
if(v==fa) continue;
dfs(v,u);
s[u]+=s[v]; //s[u]=sum(s[u的所以子节点])
for(int j=s[u];j>0;j--) //基础的背包dp
for(int k=1;k<=s[v]&&k<=j;k++)
f[u][j]=max(f[u][j],f[u][j-k]+f[v][k]+a[u][v]);
}
}
int main()
{
memset(f,-0x3f,sizeof f);
int n,m;
scanf("%d %d",&n,&m);
for(int i=1;i<n;i++) //建图
{
int u,v,w;
scanf("%d %d %d",&u,&v,&w);
h[u].push_back(v);
h[v].push_back(u);
a[u][v]=a[v][u]=w;
}
dfs(1,-1); //通过递归的方式dp
printf("%d\n",f[1][m+1]);
return 0;
}