0为根节点
dp[0][m+1]表示 从0节点开始覆盖,覆盖m+1个点的最大值
#include<stdio.h>
#include<string.h>
#include<vector>
#include<algorithm>
using namespace std;
int dp[220][220],v[220];
vector<int> q[220];
void dfs(int p,int M)
{
int len=q[p].size();
dp[p][1]=v[p];
for(int i=0;i<len;i++){ //分组背包,len组
if(M>1)
dfs(q[p][i],M-1); //递归
for(int j=M;j>=1;j--)
for(int k=1;k<j;k++)
dp[p][j]=max(dp[p][j],dp[p][j-k]+dp[q[p][i]][k]);
}
}
int main()
{
int n,m;
int i,j,k;
while(scanf("%d%d",&n,&m),n||m){
for(i=0;i<=n;i++)
q[i].clear();
memset(dp,0,sizeof(dp));
for(i=1;i<=n;i++){
scanf("%d%d",&k,&v[i]);
q[k].push_back(i);
}
v[0]=0;
dfs(0,m+1); // dfs(节点,此节点一下的将要覆盖的点) 因为算了0节点,所以m+1
printf("%d\n",dp[0][m+1]);
}
}
***************
好像 树形dp能用二维写。。
这题就是二维优化的,
然后用同样的方法优化了下,这题。。
上面的代码125ms
下面的代码15ms
#include<stdio.h>
#include<string.h>
#include<vector>
#include<algorithm>
using namespace std;
int dp[220][220],v[220];
vector<int> q[220];
int flag[220];
void dfs(int p,int M)
{
int i,j,k,u;
int len=q[p].size();
for(i=0;i<len;i++)
{
u=q[p][i];
if(flag[u])
{
for(j=M-1;j>=0;j--)
dp[u][j]=dp[p][j];
if(M-1>=0)
dfs(u,M-1);
for(j=M;j>=1;j--)
dp[p][j]=max(dp[p][j],dp[u][j-1]+v[u]);
}
else
{
for(j=M;j>=1;j--)
dp[p][j]=max(dp[p][j],dp[p][j-1]+v[u]);
}
}
}
int main()
{
int n,m;
int i,j,k;
while(scanf("%d%d",&n,&m),n||m){
for(i=0;i<=n;i++)
q[i].clear();
memset(flag,0,sizeof(flag));
memset(dp,0,sizeof(dp));
for(i=1;i<=n;i++){
scanf("%d%d",&k,&v[i]);
q[k].push_back(i);
flag[k]=1;
}
v[0]=0;
dfs(0,m);
printf("%d\n",dp[0][m]);
}
}