題目描述 Description

Ural大學有N個職員，編號為1~N。他們有從屬關系，也就是說他們的關系就像一棵以校長為根的樹，父結點就是子結點的直接上司。每個職員有一個快樂指數。現在有個周年慶宴會，要求與會職員的快樂指數最大。但是，沒有職員願和直接上司一起與會。

輸入描述 Input Description

第一行一個整數N。(1<=N<=6000)
接下來N行，第i+1行表示i號職員的快樂指數Ri。(-128<=Ri<=127)
接下來N-1行，每行輸入一對整數L,K。表示K是L的直接上司。
最後一行輸入0,0。

輸出描述 Output Description

輸出最大的快樂指數。

樣例輸入 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
【解析】
找到根結點。對於每個結點我們在去或者不去當中取最優。

如果當前結點去，當前結點去的最優值+兒子結點不去的最優值。

如果當前結點不去，兒子結點可能去也可能不去。在這二者中取一個最優值。

【代碼】

//沒有上司的舞會 
#include<iostream>
#include<cstdio>
#include<vector>
using namespace std;
#define
 
 N 6009
vector<int>vec[N];
int f[N],g[N],a[N],dad[N];
int n,l,k,rt;
inline void dfs(int x)
{
g[x]=a[x];
for(int i=0;i<vec[x].size();i++)
{
dfs(vec[x][i]);
g[x]+=f[vec[x][i]];
f[x]+=max(f[vec[x][i]],g[vec[x][i]]);
}
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
 
for(int i=1;i<=n;i++)
{
scanf("%d%d",&l,&k);
if(l==0&&k==0);
else
{
dad[l]=k;
vec[k].push_back(l);
}
}
for(rt=1;rt<=n;rt++)if(!dad[rt])break;
dfs(rt);
printf("%d",max(g[rt],f[rt]));
return 0;
}

沒有上司的舞會