hdu 1561 The more, The Better----树形dp+有依赖的背包

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能用二维写。。

http://blog.csdn.net/u010489389/article/details/19005771

这题就是二维优化的，

然后用同样的方法优化了下，这题。。

上面的代码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]);  
    }  
}