二叉苹果树
题目描述
有一棵苹果树,如果树枝有分叉,一定是分2叉(就是说没有只有1个儿子的结点)
这棵树共有N个结点(叶子点或者树枝分叉点),编号为1-N,树根编号一定是1。
我们用一根树枝两端连接的结点的编号来描述一根树枝的位置。下面是一颗有4个树枝的树
现在这颗树枝条太多了,需要剪枝。但是一些树枝上长有苹果。
给定需要保留的树枝数量,求出最多能留住多少苹果。
输入格式
第1行2个数,N和Q(1<=Q<= N,1<N<=100)。
N表示树的结点数,Q表示要保留的树枝数量。接下来N-1行描述树枝的信息。
每行3个整数,前两个是它连接的结点的编号。第3个数是这根树枝上苹果的数量。
每根树枝上的苹果不超过30000个。
输出格式
一个数,最多能留住的苹果的数量。
树形背包的基础题;
总的解法没有什么变化,就是要注意一个细节,就是如果要取 i 结点的子树,那么结点 i 与这个子树的连接边 i–j 就必须取;
所以状态转移方程为:dp[u][j]=max{dp[u][j],dp[u][j-k-1]+dp[v][k]+w[i]};
这个减 1 就是关键,这个要和有体积和价值的树上背包联合理解;
#include<bits/stdc++.h>
#define LL long long
#define pa pair<int,int>
#define ls k<<1
#define rs k<<1|1
#define inf 0x3f3f3f3f
using namespace std;
const int N=110;
const int M=2000100;
const int mod=1e9;
int n,m,head[N],cnt;
int dp[N][N];
struct Node{
int to,nex,w;
}edge[N*2];
void add(int p,int q,int w){
edge[cnt].w=w;
edge[cnt].to=q;
edge[cnt].nex=head[p];
head[p]=cnt++;
}
void dfs(int p,int q){
for(int i=head[p];~i;i=edge[i].nex){
int v=edge[i].to;
int w=edge[i].w;
if(v!=q){
dfs(v,p);
for(int j=m;j>=1;j--){
for(int k=0;k<j;k++){
dp[p][j]=max(dp[p][j],dp[p][j-k-1]+dp[v][k]+w);
}
}
}
}
}
int main(){
memset(head,-1,sizeof(head));
scanf("%d%d",&n,&m);
for(int i=1;i<n;i++){
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
add(a,b,c),add(b,a,c);
}
dfs(1,-1);
cout<<dp[1][m]<<endl;
return 0;
}