描述
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
程序
#include<iostream>
#include<cstdio>
#include<vector>
using namespace std;
vector<int> son[10010];
int n;
int f[10010][1],h[10010];
bool v[10010];
void dp(int x){
f[x][0]=0;
f[x][1]=h[x];
for (int i=0;i<son[x].size();i++){
int y=son[x][i];
if (y<0) return;
dp(y);
f[x][0]+=max(f[y][0],f[y][1]);
f[x][1]+=f[y][0];
}
}
int main(){
cin>>n;
for (int i=1;i<=n;i++) cin>>h[i];
for (int i=2;i<=n;i++){
int m;
cin>>m;
v[i]=1;
son[m].push_back(i);
}
dp(1);
cout<<max(f[1][0],f[1][1])<<endl;
return 0;
}
注释
这是正解,但不知道为什么错了QAQ