比之前树形背包要简单多了,没有背包的体积限制
dp[ i ] [ 0 ] 以i为根(不包括i本身)的树上取到的最大值
dp[ i ] [ 1 ] 以i为根(包括i本身)的树上取到的最大值
具体代码如下:
#include
#include
#include
#include
using namespace std;
int N,V[6100];
vectorson[6010];
int dp[6100] ,vis[6100];
void dfs(int n){
vis[n]=1;
int len=son[n].size();
dp[n] =V[n];
for(int i=0;i
int k=son[n][i];
if(vis[k]) continue;
dfs(k);
if(dp[n]
dp[n][0]=max(dp[n][0],max(dp[n][0]+dp[k] ,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] );
// printf("%d %d\n",dp[i][0],dp[i] );
printf("%d\n",max(dp[i][0],dp[i] ));
}
}
return 0;
}
/*
7
1
-1
-1
-1
-1
-1
-1
1 3
2 3
6 4
7 4
4 5
3 5
0 0
*/