vijos1706 舞会 (树形dp)

一道不知道是不是树形dp的水题。。

就是选了一个点不能选他的父亲和儿子。 求一个最大值

见代码

描述

Arthur公司是一个等级森严的公司，它们有着严格的上司与下属的关系，公司以总 裁为最高职位，他有若干个下属，他的下属又有若干个下属，他的下属的下属又有若干个下属……现接近年尾，公司组织团拜活动，活动中有一部分是自由舞会，公 司的每个职员都有一个搞笑值，现要你制定一套哪些人上台的方案，使得台上所有演员的搞笑值最大。当然，职员们是不会和他们的顶头上司一起上台的。

格式

输入格式

第一行一个整数N，表示这个公司总共的职员个数。

接下来一行有N个整数，由空格隔开，第i个整数表示职员i的搞笑值Ai（-1327670≤Ai≤1327670）。

接下来N-1行，每行一个1到N的整数，第i个整数表示职员i+1的顶头上司是谁，当然总裁就是职员1。

输出格式

一个整数，表示台上所有职员搞笑值之和的最大值。

样例1

样例输入1[复制]

7
1 1 1 1 1 1 1
1
1
5
1
4
4

样例输出1[复制]

5

提示

【数据范围】
保证100%的数据N≤5000

来源

URAL 由xzx改编

 

 

代码：

#include <iostream>
#include <vector>
#include <cstdio>
using namespace std;
#define MAX 5000+10
vector<int> son[MAX];
int fun[MAX];
int N;
int dp[MAX][2];
int dfs(int i,int part)
{
    int& d =dp[i][part];
    if(d)return d;

    if( part==0)
    {
        for(int k=0;k<son[i].size();k++)
        {
            d += max(dfs( son[i][k] , 1), dfs(son[i][k],0));
        }
        return d ;
    }
    if( part == 1)
    {
        for(int k=0;k<son[i].size();k++)
        {
            d += dfs(son[i][k],0);
        }
        return d= d+fun[i];
    }
}
int main()
{
    scanf("%d",&N);
    for(int i=1;i<=N;i++)
    {
        scanf("%d",&fun[i]);
    }
    for(int i=2;i<=N;i++)
    {
        int d;
        scanf("%d",&d);
        son[d].push_back(i);
    }
    printf("%d
",max(dfs( 1,0),dfs(1,1)));
    return 0;
}