hdu 1011 Starship Troopers(树形背包)

版权声明:本文为博主原创文章,欢迎转载。 https://blog.csdn.net/define_danmu_primer/article/details/53897536

题意

一棵树,花费一定,使得获得的价值最大

思路

规范的树形背包,过程不是很理解。

代码

#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
using namespace std;
const int Ni=110;
int n,m;
int coss[Ni],weg[Ni];
int dp[Ni][Ni];
bool vis[Ni];
vector<int> dv[Ni];
void dfs(int p){
    int i,j,k;
    int tmp=(coss[p]+19)/20;
    for(i=tmp;i<=m;++i) dp[p][i]=weg[p];
    vis[p]=1;
    for(i=0;i<dv[p].size();++i){
        int t=dv[p][i];
        if(vis[t]) continue;
        dfs(t);
        for(j=m;j>=tmp;--j){
            for(k=1;k<=j-tmp;++k)
                dp[p][j]=max(dp[p][j],dp[p][j-k]+dp[t][k]);
        }
    }
}
int main(){
    int i;
    while(scanf("%d%d",&n,&m)&&n!=-1){
        for(i=0;i<=n;++i) dv[i].clear();
        memset(dp,0,sizeof dp);
        memset(vis,0,sizeof vis);
        for(i=1; i<=n; i++)
            scanf("%d%d",coss+i,weg+i);
        for(i=1;i<n;++i){
            int u,v;
            scanf("%d%d",&u,&v);
            dv[u].push_back(v);
            dv[v].push_back(u);
        }
        if(m==0){
            printf("0\n");
            continue;
        }
        dfs(1);
        printf("%d\n",dp[1][m]);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/define_danmu_primer/article/details/53897536