Ural1018 二叉苹果树
题目
有一棵苹果树,如果树枝有分叉,一定是分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个。
输出格式
一个数,最多能留住的苹果的数量。
样例输入
5 2
1 3 1
1 4 10
2 3 20
3 5 20
样例输出
21
方法:树形DP,详细参见关联文章
给出代码:
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int MAXN=103;
typedef struct
{
int lc,rc;
int val;
}Node;
Node tree[MAXN];
int m,n;
int arr[MAXN][MAXN];
int f[MAXN][MAXN];
int vis[MAXN];
void Create(int root)
{
vis[root]=1;
for(int i=1;i<=m;i++)
{
if(!vis[i]&&arr[root][i])
{
if(!tree[root].lc)
tree[root].lc=i;
else tree[root].rc=i;
tree[i].val=arr[root][i];
Create(i);
}
}
}
int TreeDP(int root,int e)
{
if(!root||!e)return 0;
if(f[root][e])return f[root][e];
int maxx=0;
int l,r;
for(int i=0;i<e;i++)
{
l=TreeDP(tree[root].lc,i);
r=TreeDP(tree[root].rc,e-i-1);
maxx=max(l+r,maxx);
}
f[root][e]=maxx+tree[root].val;
return f[root][e];
}
int main()
{
memset(arr,0,sizeof(arr));
memset(vis,0,sizeof(vis));
memset(tree,0,sizeof(tree));
scanf("%d %d",&m,&n);
int l,f,v;
for(int i=1;i<m;i++)
{
scanf("%d %d %d",&l,&f,&v);
arr[l][f]=arr[f][l]=v;
}
Create(1);
printf("%d\n",TreeDP(1,++n));
return 0;
}