【SSL 1605&洛谷 P2015】二叉苹果树【树形DP】

题目描述

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

分析：

由于儿子节点的选择影响其父节点 可选用树形DP
构建一棵树以及连双向边
动态能量转移方程：

f[dep][j]=max(f[dep][j],f[a[i].to][k]+f[dep][j-k-1]+SCP[dep][a[i].to]);

其中 $f[i][j]$ 代表以$i$为父节点保留$j$条枝最多苹果数……
$a[i].to$表示$dep$的子节点……
$j-k$表示 $a[i].to$的兄弟节点(同父的分支)
最后再 $-1$是因为子连父还需一条……

CODE：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
struct node{
	int to,next;
}a[210];
int tot,n,q,x,y,v,ans,h[105],e[105],f[105][105],SCP[105][105];
void add(int x,int y)
{
	a[++tot]=(node){y,h[x]}; 
	h[x]=tot;
}
void Tree_dp(int dep){  //树形DP
	e[dep]=1;
	for(int i=h[dep];i;i=a[i].next)
	{
		if(e[a[i].to]==1) continue;  //不重复搜父节点
		Tree_dp(a[i].to);
		for(int j=q;j>0;j--)
			for(int k=j-1;k>=0;k--)
				f[dep][j]=max(f[dep][j],f[a[i].to][k]+f[dep][j-k-1]+SCP[dep][a[i].to]);
				//动态能量转移方程
	}
} 
int main(){
	scanf("%d%d",&n,&q);
	for(int i=1;i<n;i++)
	{
		scanf("%d%d%d",&x,&y,&v);
		SCP[x][y]=SCP[y][x]=v;  //记录权值
		add(x,y);
		add(y,x);  //建立双向邻接表
	}
	Tree_dp(1);
	printf("%d",f[1][q]);
	return 0;
}