二叉苹果树【洛谷P2015】

题目描述

有一棵苹果树，如果树枝有分叉，一定是分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;
}