陶陶吃苹果
背景
陶陶很喜欢吃苹果。
描述
curimit知道陶陶很喜欢吃苹果。于是curimit准备在陶陶生日的时候送给他一棵苹果树。
curimit准备了一棵这样的苹果树作为生日礼物:这棵苹果树有n个节点,每个节点上有c[i]个苹果,这棵树高度为h。
可是,当curimit把这棵树给陶陶看的时候,陶陶却说:“今年生日不收礼,收礼只收节点数减高度不超过k的苹果树。”这下curimit犯难了,curimit送来的树枝繁叶茂,不满足节点数-高度≤k。于是curimit决定剪掉一些枝条,使得修剪过后的树满足节点数-高度≤k,但是curimit又想保留尽量多的苹果数目。curimit想请你帮他算算经过修剪后的树最多能保留多少个苹果。
注:
一, 节点1为树根,不能把它剪掉。
二, 1个节点的树高度为1。
格式
输入格式
输入文件的第一行为两个整数n,k分别表示这棵树有n个节点,修剪后的树节点数-高度≤k。
第二行开始到第n+1行,每行有两个数,第i+1行的两个数father[i]和c[i]分别表示节点i的父亲是father[i]和节点i处有c[i]个苹果。
规定:节点1的父亲为0。
输出格式
输出文件仅包含一行,ans,表示在满足修建后的树节点数-高度≤k的条件下,最多能保留多少个苹果。
样例1
样例输入1
5 1
0 1
1 1
1 3
2 10
3 4
Copy
样例输出1
15
Copy
限制
全部1秒
提示
【约定】
对于10%的数据, n≤10
对于30%的数据, n≤100 0≤k≤20
对于100%的数据,n≤4000 0≤k≤=500
来源
思路: 树形dp
代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N =4005;
ll dp1[N][505];
ll dp2[N][505];
ll sum[N];
ll c[N];
int rt;
vector< int >ve[N];
int n,k,m;
void dfs1(int u)
{
for(int i=0;i<ve[u].size();i++){
int v=ve[u][i];
sum[v]=sum[u]+c[v];
for(int j=0;j<=m;j++) dp1[v][j]=dp1[u][j];
dfs1(v);
for(int j=1;j<=m;j++) dp1[u][j]=max(dp1[u][j],dp1[v][j-1]+c[v]); //依赖性体现在这个方程中
}
}
void dfs2(int u)
{
for(int i=ve[u].size()-1;i>=0;i--){
int v=ve[u][i];
sum[v]=sum[u]+c[v];
for(int j=0;j<=m;j++) dp2[v][j]=dp2[u][j];
dfs2(v);
for(int j=1;j<=m;j++) dp2[u][j]=max(dp2[u][j],dp2[v][j-1]+c[v]);
}
}
int main()
{
int u;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%d %lld",&u,&c[i]);
if(!u){
rt=i;
}
else{
ve[u].push_back(i);
}
}
sum[rt]=c[rt];
dfs1(rt);
dfs2(rt);
ll ans=0;
for(int i=1;i<=n;i++){
if(ve[i].size()==0){
for(int j=0;j<=m;j++){
ans=max(ans,dp1[i][j]+dp2[i][m-j]+sum[i]);
}
}
}
cout<<ans<<endl;
return 0;
}