poj 2486

来源：互联网发布：汉客拉杆箱怎么样知乎编辑：程序博客网时间：2024/06/16 07:34

dp[0][i][j]表示以i为根节点步行j步，且最终回到i，可以得到的最大苹果数量。dp[1][i][j]表示最终不回到i可以得到的最大苹果数量。

#include <cstdio>#include <cstring>#include <algorithm>#include <vector>#include <map>using namespace std;#define  N 221#define  inf 0x7f7f7f7fint dp[2][N][N];int root;int n,k;int cost[111];vector<int>g[N];int vis[N];void dfs(int s){    for(int i=0;i<=k;i++){        dp[0][s][i]=dp[1][s][i]=cost[s];//初始化    }    vis[s] = 1;    for(int i=0;i<g[s].size();i++){        int e = g[s][i];        if(!vis[e]){            dfs(e);            for(int j = k;j >= 0;j--){                for(int h=0;h <= j;h++){                    dp[0][s][j+2]=max(dp[0][s][j+2],dp[0][e][h]+dp[0][s][j-h]);                    dp[1][s][j+2]=max(dp[1][s][j+2],dp[0][e][h]+dp[1][s][j-h]);                    dp[1][s][j+1]=max(dp[1][s][j+1],dp[1][e][h]+dp[0][s][j-h]);                }            }        }    }}int main(){    while(scanf("%d%d",&n,&k)!=EOF){        memset(dp,0,sizeof(dp));        memset(vis,0,sizeof(vis));        memset(cost,0,sizeof(cost));        for(int i=1;i<=n;i++){            g[i].clear();            scanf("%d",&cost[i]);        }        for(int i=1;i<n;i++){            int s,e;            scanf("%d%d",&s,&e);            g[s].push_back(e);            g[e].push_back(s);        }        dfs(1);        printf("%d\n",dp[1][1][k]);    }}

0 0