hdu1520 Anniversary party 树形dp

来源：互联网发布：淘宝主图的尺寸编辑：程序博客网时间：2024/05/16 08:51

hdu1520

比之前树形背包要简单多了，没有背包的体积限制

dp[ i ] [ 0 ] 以i为根（不包括i本身）的树上取到的最大值

dp[ i ] [ 1 ] 以i为根（包括i本身）的树上取到的最大值

#include<iostream>#include<cstdio>#include<cstring>#include<vector>using namespace std;int N,V[6100];vector<int>son[6010];int dp[6100][2],vis[6100];void dfs(int n){vis[n]=1;int len=son[n].size();dp[n][1]=V[n];for(int i=0;i<len;i++){intk=son[n][i];if(vis[k])continue;dfs(k);if(dp[n][1]<dp[n][1]+dp[k][0])dp[n][1]=dp[n][1]+dp[k][0];dp[n][0]=max(dp[n][0],max(dp[n][0]+dp[k][1],dp[n][0]+dp[k][0]));}}int main(){int i,j,k,u,v;while(scanf("%d",&N)!=EOF){memset(vis,0,sizeof(vis));memset(dp,0,sizeof(dp));for(i=1;i<=N;i++){scanf("%d",&V[i]);son[i].clear();}while(scanf("%d%d",&u,&v),u||v){son[v].push_back(u);vis[u]=1;//非根节点 }for(i=1;i<=N;i++)if(vis[i]==0){memset(vis,0,sizeof(vis));//printf("%d\n",i);dfs(i);//for(j=1;j<=N;j++)printf("%d    %d %d\n",j,dp[j][0],dp[j][1]);//printf("%d %d\n",dp[i][0],dp[i][1]);printf("%d\n",max(dp[i][0],dp[i][1]));}}return 0;}/*71-1-1-1-1-1-11 32 36 47 44 53 50 0*/