有一棵苹果树,如果树枝有分叉,一定是分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
题解:
树形背包模板题,考虑到每次删减最小的边有后效性,无法得出正解可以考虑背包
f[u][i]表示以u为根节点的拥有i条边的子树拥有的最大苹果数量。
显然转移方程可以写成:
f[u][i] = max(f[u][i], f[u][i - j + 1] + f[v][j] + w[i])
接着直接套树形背包的模板即可。
#include <bits/stdc++.h>
#define dbug(x) cout << #x << "=" << x << endl
#define eps 1e-8
#define pi acos(-1.0)
using namespace std;
typedef long long LL;
const int inf = 0x3f3f3f3f;
template<class T>inline void read(T &res)
{
char c;T flag=1;
while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;res=c-'0';
while((c=getchar())>='0'&&c<='9')res=res*10+c-'0';res*=flag;
}
const int maxn = 2e5 + 7;
int n, q;
int cnt, w[maxn], head[maxn], nxt[maxn], edge[maxn];
int ans = 0;
int f[107][107];
void BuildGraph(int u, int v, int c) {
++cnt;
edge[cnt] = v;
nxt[cnt] = head[u];
head[u] = cnt;
w[cnt] = c;
}
void dfs(int u, int fa) {
for ( int i = head[u]; i; i = nxt[i] ) {
int v = edge[i];
if(v == fa) {
continue;
}
dfs(v, u);
for ( int j = q; j; --j ) {
for ( int k =j - 1; k >= 0; --k ) {
f[u][j] = max(f[u][j], f[u][j - k - 1] + f[v][k] + w[i]);
// printf("f[%d][%d]:%d\n",u, j, f[u][j]);
}
}
}
}
int main()
{
read(n); read(q);
for ( int i = 1; i <= n - 1; ++i ) {
int u, v, w;
read(u); read(v); read(w);
BuildGraph(u, v, w);
BuildGraph(v, u, w);
}
dfs(1, 1);
for ( int i = 1; i <= n; ++i ) {
ans = max(f[i][q], ans);
}
ans = f[1][q];
cout << ans << endl;
return 0;
}
京公网安备 11010502036488号