hdu 1011 Starship Troopers (树形dp+背包)

最新推荐文章于 2024-05-22 17:32:19 发布

原创最新推荐文章于 2024-05-22 17:32:19 发布 · 98 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种解决背包问题的方法，特别适用于具有树状结构的数据。通过递归地使用深度优先搜索（DFS），该算法有效地计算了每个节点的最佳价值，确保了在给定的预算下获得最大的总价值。

#include <iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#define N 105
using namespace std;

int c[N],w[N],dp[N][N],m;
vector<int>edge[N];

void dfs(int u,int fa)
{
    int v,len=edge[u].size(),i,j,k;
    for(j=m;j>=c[u];j--) dp[u][j]=w[u];
	//for(j=m;j>=c[u] && j>0;j--) dp[u][j]=w[u];
    for(i=0;i<len;i++){
        v=edge[u][i];
        if(v==fa) continue;
        dfs(v,u);
        for(j=m;j>c[u];j--){
            for(k=1;k<=m && k<=j-c[u];k++){ //每个儿子节点都至少要派一个人去，所以至少为1
                dp[u][j]=max(dp[u][j],dp[u][j-k]+dp[v][k]);
            }
        }
    }
}

int main()
{
    int n;
    while(~scanf("%d%d",&n,&m)){
		if(n==-1 && m==-1) break;
        int i;
        memset(dp,-32,sizeof(dp));
        int INF=dp[0][0];
        for(i=0;i<=n;i++) edge[i].clear();
        for(i=1;i<=n;i++){
            scanf("%d%d",c+i,w+i);
            c[i]=(c[i]%20)?(c[i]/20+1):(c[i]/20);
        }
        for(i=1;i<n;i++){
            int a,b;
            scanf("%d%d",&a,&b);
            edge[a].push_back(b);
            edge[b].push_back(a);
        }
		if(m==0){
			printf("0\n");
			continue;
		}
		dfs(1,0);
		if(dp[1][m]==INF) puts("0");
		else printf("%d\n",dp[1][m]);
    }
    return 0;
}