HDU 1011 Starship Troopers(树形DP入门题)

来源:互联网 发布:网络rtt 编辑:程序博客网 时间:2024/04/27 22:52

初学者,应该看题解也是懵懵懂懂的吧。这题网上题解很多,多找几篇总会找到合适的。

#include<iostream>#include<cstdio>#include<vector>using namespace std;const int maxn=105;int dp[maxn][maxn];int c[maxn],w[maxn];int n,m;vector<int>G[maxn];void dfs(int u,int fa){    int cost;    if(c[u]%20) cost=c[u]/20+1;    else cost=c[u]/20;    for(int i=cost;i<=m;i++) dp[u][i]=w[u];    for(int i=0;i<G[u].size();i++)    {        int v=G[u][i];        if(v==fa) continue;        dfs(v,u);        for(int j=m;j>=cost;j--)//必须从大到小逆序更新,知道0-1背包二维数组优化成一维的应该很快就能明白            for(int k=1;k+j<=m;k++)            if(dp[v][k]) dp[u][j+k]=max(dp[u][j+k],dp[u][j]+dp[v][k]);    }}int main(){    while(~scanf("%d%d",&n,&m),n+m!=-2)    {        for(int i=0;i<=n;i++) G[i].clear();        fill(&dp[0][0],&dp[maxn][0],0);        for(int i=1;i<=n;i++) scanf("%d%d",&c[i],&w[i]);        for(int i=1;i<n;i++)        {            int from,to;            scanf("%d%d",&from,&to);            G[from].push_back(to);            G[to].push_back(from);        }        if(m==0) {puts("0");continue;}//这是个坑点        dfs(1,-1);        printf("%d\n",dp[1][m]);    }    return 0;}



0 0