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

来源：互联网发布：mac魔兽世界编辑：程序博客网时间：2024/05/22 15:53

树形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 105using 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