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

8 篇文章 0 订阅
5 篇文章 0 订阅

树形dp加泛化背包,状态转移方程dp[u][j]=max(dp[u][j],dp[v][k]+dp[u][j-k]),有很多细节需要注意,叶子节点就算没有bug也要留人,还有m=0的特殊情况。

#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++){ 
                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;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值