Problem Description
ACboy很喜欢玩一种战略游戏,在一个地图上,有N座城堡,每座城堡都有一定的宝物,在每次游戏中ACboy允许攻克M个城堡并获得里面的宝物。但由于地理位置原因,有些城堡不能直接攻克,要攻克这些城堡必须先攻克其他某一个特定的城堡。你能帮ACboy算出要获得尽量多的宝物应该攻克哪M个城堡吗?
Input
每个测试实例首先包括2个整数,N,M.(1 <= M <= N <= 200);在接下来的N行里,每行包括2个整数,a,b. 在第 i 行,a 代表要攻克第 i 个城堡必须先攻克第 a 个城堡,如果 a = 0 则代表可以直接攻克第 i 个城堡。b 代表第 i 个城堡的宝物数量, b >= 0。当N = 0, M = 0输入结束。
Output
对于每个测试实例,输出一个整数,代表ACboy攻克M个城堡所获得的最多宝物的数量。
Sample Input
3 2
0 1
0 2
0 3
7 4
2 2
0 1
0 4
2 1
7 1
7 6
2 2
0 0
Sample Output
5
13
解法:
依赖关系可以转化为一颗树,
令的d[i][j]为点i为根的子树中选择j个的方案数,显然d[i][1]=a[i]
转移方程:
for(int i=m;i>=1;i--){//逆向枚举保证无后效性,枚举x中选择的个数
for(int j=1;j<=i-1;j++){//枚举v中选择的个数,因为x必选,所以是<=i-1
d[x][i]=max(d[x][i],d[v][i-j]+d[f][j]);
}
}
因为子树中可能没有m个点,因此还可以从min(sz[x],m)开始枚举,稍微优化一下
这题是森林,可以建立虚根0变成一颗树,然后从0号点开始dp
注意选0号点是不消耗次数的,稍微判断一下就行了
code:
#include<bits/stdc++.h>
using namespace std;
const int maxm=205;
vector<int>g[maxm];
int d[maxm][maxm];
int sz[maxm];
int a[maxm];
int n,m;
void dfs(int x){
d[x][1]=a[x];
sz[x]=1;
for(int v:g[x]){
dfs(v);
sz[x]+=sz[v];
}
for(int v:g[x]){
for(int s=min(m,sz[x])+(x==0);s>=1;s--){//枚举x中选择的个数,0点+1,因为选0点不消耗次数
for(int j=1;j<=min(s-1,sz[v]);j++){//枚举v中选择的个数,因为x必选,因此里面是s-1
d[x][s]=max(d[x][s],d[x][s-j]+d[v][j]);
}
}
}
}
signed main(){
while(cin>>n>>m){
if(n+m==0)break;
for(int i=0;i<=n;i++)g[i].clear();
for(int i=1;i<=n;i++){
int x;cin>>x>>a[i];
g[x].push_back(i);
}
memset(d,0,sizeof d);
dfs(0);
cout<<d[0][m+1]<<endl;
}
return 0;
}