当前位置: 代码迷 >> 综合 >> 【SSL 1605&洛谷 P2015】二叉苹果树【树形DP】
  详细解决方案

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

热度:37   发布时间:2024-02-10 17:19:20.0

题目描述

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