题目描述
题目
有一棵苹果树,如果树枝有分叉,一定是分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是因为子连父还需一条……
SCP记录的是权值……
好吧跑的挺快的 0MS
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){ //树形DPe[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;
}