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

前端之家收集整理的这篇文章主要介绍了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,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://www.jb51.cc/article/p-msnhrvnr-xx.html

这题就是二维优化的,

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

上面的代码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,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,k;  
    while(scanf("%d%d",n||m){  
        for(i=0;i<=n;i++)  
            q[i].clear();  
		memset(flag,sizeof(flag));
        memset(dp,sizeof(dp));  
        for(i=1;i<=n;i++){  
            scanf("%d%d",&v[i]);  
            q[k].push_back(i);  
			flag[k]=1;
        }  
        v[0]=0;  
        dfs(0,m);          
        printf("%d\n",dp[0][m]);  
    }  
}  

猜你在找的设计模式相关文章