树形dp

洛谷P1352------树形dp

题目链接：https://www.luogu.org/problem/P1352#submit

dp思路：
第一步：确定状态
用dp[i][0]表示不选择i点时，i点及其子树能选出的最多快乐指数，dp[i][1]表示选择i点时，i点及其子树的最多快乐指数

第二步：确定状态转移方程
dp[i][0] = Σ(max (dp[j][0], dp[j][1]))
dp[i][1] = r[i]+ Σdp[j][0]
(j是i的儿子)
(r为对应的快乐指数)

初始化（边界条件）
dp[i][0] = 0, dp[i][1] = r[i]

结果为max(dp[root][0], dp[root][1])

实现dp递推的过程是依靠记忆化搜索（实际上就是一个树的后序遍历）

AC代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<string>
#include<stack>
#include<map>
#include<queue>
#include<set>
#include<cmath>
#include<cstdlib>
//#include<bits/stdc++.h>
using namespace std;
#define scan(n) scanf("%d",&n)
#define scand(n) scanf("%lf",&n)
#define ll long long

int n;
int r[6005];
vector<int> son[6005];
int dp[6005][2];
int ro[6005];
void dfs(int x)
{
    dp[x][0]=0;
    dp[x][1]=r[x];
    int s;
    for(int i=0;i<son[x].size();i++)
    {
        s=son[x][i];
        dfs(s);
        dp[x][0]+=max(dp[s][1],dp[s][0]);
        dp[x][1]+=dp[s][0];
    }
    return;
}

int main()
{
    int i;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&r[i]);
    int u,v;//v是u的上司
    memset(ro,0,sizeof(ro));
    for(i=1;i<=n-1;i++)
    {
        scanf("%d%d",&u,&v);
        son[v].push_back(u);
        ro[u]=1;
    }
    getchar();
    getchar();
    int root;
    for(int i=1;i<=n;i++)
    {
        if(ro[i]==0)
        {
            root=i;
            break;
        }
    }
    dfs(root);
    printf("%d",max(dp[root][0],dp[root][1]));
}