【题解】没有上司的晚会

Description

有个公司要举行一场晚会。
为了能玩得开心，公司领导决定：如果邀请了某个人，那么一定不会邀请他的上司
（上司的上司，上司的上司的上司……都可以邀请）。
每个参加晚会的人都能为晚会增添一些气氛，求一个邀请方案，使气氛值的和最大。

Input

第1行一个整数N（1<=N<=6000）表示公司的人数。
接下来N行每行一个整数。第i行的数表示第i个人的气氛值x(-128<=x<=127)。
接下来每行两个整数L，K。表示第K个人是第L个人的上司。
输入以0 0结束。

Output

一个数，最大的气氛值和。

Sample Input

7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
0 0

Sample output

5

解题思路

【树形DP经典题型】dp【i】【0】dp【i】【1】表示从i取或者不取最大值。

dp【i】【0】+=max（dp【son】【1】，dp【son】【0】）；

dp【i】【1】+=max（dp【son】【0】）；

Hint

#include<bits/stdc++.h>
using namespace std;
int n,cnt,ans;
int s[1000000],v[1000000];
vector<int> tot[100000];
int dp[100000][10];
void f(int x){
dp[x][0]=0;
dp[x][1]=s[x];
for(int i=0;i<tot[x].size();i++){
int y=tot[x][i];
f(y);
dp[x][0]+=max(dp[y][0],dp[y][1]);
dp[x][1]+=dp[y][0];
}
}
int main(){
cin>>n;
for(int i=1;i<=n;i++){
cin>>s[i];
}
for(int i=1;i<=n-1;i++){
int a,b;
cin>>a>>b;
tot[b].push_back(a);
v[a]=1;
}
for(int i=1;i<=n;i++){
if(!v[i]){
cnt=i;
break;
}
}
f(cnt);
ans=max(dp[cnt][0],dp[cnt][1]);
cout<<ans;
return 0;
}