舞会

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

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

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

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

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

样例输入
7
1 1 1 1 1 1 1
1
1
5
1
4
4
样例输出
5
提示
保证100%的数据N≤5000

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#define inf 100000000
#define ll long long
#define mo 1000000007
using namespace std;
int dp[5005][2],a[5005],Head[5005],ret[5005],Next[5005];
int u,n,tot;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
void ins(int u,int v){tot++;ret[tot]=v;Next[tot]=Head[u];Head[u]=tot;}
int dfs(int u,int f)
{
    if (dp[u][f]!=-inf) return dp[u][f];
    dp[u][f]=0;
    if (f) dp[u][f]+=a[u];
    for (int i=Head[u];i;i=Next[i])
    {
        int v=ret[i];
        if (f) dp[u][f]+=dfs(v,0);
        else dp[u][f]+=max(dfs(v,0),dfs(v,1));
    }
    return dp[u][f];
}
int main()
{
    n=read();
    for (int i=1;i<=n;i++) a[i]=read(),dp[i][1]=dp[i][0]=-inf;
    for (int i=2;i<=n;i++)
    {
        u=read();ins(u,i);
    }
    int ans=max(dfs(1,1),dfs(1,0));
    printf("%d",ans);
    return 0;
}