HDU 1520 Anniversary party 树形dp-优快云博客

本文介绍了一种使用状态转移方程实现的DP算法，通过递归深度优先搜索遍历树形结构来解决问题。该算法能够计算出每个节点两种状态下的最大值，即选择当前节点或跳过当前节点时的最大收益。

状态转移方程：

dp[u][0] = max(dp[v][0],dp[v][1]);

dp[u][1] += dp[v][0];

#include <iostream>
#include <cstdio>
#include <vector>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 6010;
vector<int>tree[maxn];

int rat[maxn],deg[maxn],dp[maxn][2];
int max(int x,int y){
    return x > y ? x : y;
}
void dfs(int u){
    if(0 == tree[u].size())
    {
        dp[u][1] = rat[u];
        return ;
    }
    dp[u][1] += rat[u];
    for(int i = 0; i < tree[u].size(); i++){
        int v = tree[u][i];
        dfs(v);
        dp[u][0] += max(dp[v][0],dp[v][1]);
        dp[u][1] += dp[v][0];
    }
}
int main()
{
   // freopen("in.txt","r",stdin);
    int n,l,k,root;
    while( ~scanf("%d",&n) ){
        for( int i = 1; i <= n; i++){
            tree[i].clear();
            scanf("%d",rat+i);
        }
        memset(deg,0,sizeof(deg));
        while( ~scanf("%d%d",&l,&k) , l||k){
            deg[l]++;
            tree[k].push_back(l);
        }
        memset(dp,0,sizeof(dp));
        for( root = 1; root <= n; root ++){
            if( !deg[root]){
                dfs(root);
                break;
            }
        }
        printf("%d\n",max(dp[root][0],dp[root][1]));
    }
    return 0;
}

转载于:https://www.cnblogs.com/LUO257316/p/3222195.html