poj 2486 树形DP

来源：互联网发布：dede目录权限优化工具编辑：程序博客网时间：2024/06/04 17:57

见注释。

AC代码如下：

#include <iostream>#include <cstdio>#include <cstring>#include <vector>using namespace std;int N, K;int dp[2][101][201];vector<int> edge[200];bool visit[101];int apple[101];inline int max( int a, int b ){return ( a > b ? a : b );}void DFS( int n ){if( visit[n] ){return;}visit[n] = true;for( int i = 0; i <= K; i++ ){//当不往下走时dp[0][n][i] = dp[1][n][i] = apple[n];}for( int i = 0; i < edge[n].size(); i++ ){//继续往下走 DPint t = edge[n][i];if( visit[t] ){continue;}DFS( t );for( int j = K; j >= 0; j-- ){for( int k = 0; k <= j; k++ ){dp[0][n][j+2] = max( dp[0][n][j+2], dp[0][n][j-k] + dp[0][t][k] );//在t中走k步获得的最大再回到ndp[1][n][j+1] = max( dp[1][n][j+1], dp[0][n][j-k] + dp[1][t][k] );//在t中走k步或得最大，留在t的子树或tdp[1][n][j+2] = max( dp[1][n][j+2], dp[0][t][k] + dp[1][n][j-k] );//在t中走k步或得最大，再回到n，再留在n的子树或n}}}}int main(){while( scanf( "%d%d", &N, &K ) != EOF ){memset( dp, 0, sizeof( dp ) );memset( visit, false, sizeof( visit ) );for( int i = 0; i < 200; i++ ){edge[i].clear();}for( int i = 1; i <= N; i++ ){cin >> apple[i];}for( int i = 1; i < N; i++ ){int temp1, temp2;cin >> temp1 >> temp2;edge[temp1].push_back( temp2 );edge[temp2].push_back( temp1 );}DFS(1);cout << dp[1][1][K] << endl;}return 0;}