题目描述
有一棵苹果树,如果树枝有分叉,一定是分2叉(就是说没有只有1个儿子的结点)
这棵树共有N个结点(叶子点或者树枝分叉点),编号为1-N,树根编号一定是1。
我们用一根树枝两端连接的结点的编号来描述一根树枝的位置。下面是一颗有4个树枝的树
2 5
\ /
3 4
\ /
1
现在这颗树枝条太多了,需要剪枝。但是一些树枝上长有苹果。
给定需要保留的树枝数量,求出最多能留住多少苹果。
输入格式
第1行2个数,N和Q(1<=Q<= N,1<N<=100)。
N表示树的结点数,Q表示要保留的树枝数量。接下来N-1行描述树枝的信息。
每行3个整数,前两个是它连接的结点的编号。第3个数是这根树枝上苹果的数量。
每根树枝上的苹果不超过30000个。
输出格式
一个数,最多能留住的苹果的数量。
输入输出样例
输入 #1
5 2
1 3 1
1 4 10
2 3 20
3 5 20
输出 #1
21
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
void maketree(int v);
const int N=105;
int n,q;
int tree[N][5],ma[N][N];
int num[N];
int f[N][N];
void build(int x,int y,int lor) {
num[y]=ma[x][y];
tree[x][lor]=y;
ma[x][y]=-1;
ma[y][x]=-1;
maketree(y);
}
void maketree(int v) {
int lr=0;
for(int i=0; i<=n; i++)
if(ma[v][i]>=0) {
lr++;
build(v, i, lr);
if(lr==2) {
return;
}
}
}
void dfs(int v,int k) {
if(k==0) {
f[v][k]=0;
} else if(tree[v][1]==0 && tree[v][2]==0) {
f[v][k]=num[v];
} else {
f[v][k]=0;
for(int i=0; i<k; i++) {
if(f[tree[v][1]][i]==0)
dfs(tree[v][1],i);
if(f[tree[v][2]][k-i-1]==0)
dfs(tree[v][2],k-i-1);
f[v][k]=max(f[v][k],(f[tree[v][1]][i]+f[tree[v][2]][k-i-1]+num[v]));
}
}
}
int main()
{
scanf("%d%d",&n,&q);
for(int i=0; i<=n; i++)
for(int j=0; j<=n; j++) {
ma[i][j]=-1;
ma[j][i]=-1;
}
for(int i=0; i<n-1; i++) {
int x,y,xy;
scanf("%d%d%d",&x,&y,&xy);
ma[x][y]=xy;
ma[y][x]=xy;
}
maketree(1);
dfs(1,q+1);
printf("%d\n",f[1][q+1]);
return 0;
}