没有上司的晚会
Description
Ural大学有N个职员,编号为1~N。他们有从属关系,也就是说他们的关系就像一棵以校长为根的树,父结点就是子结点的直接上司。每个职员有一个快乐指数。现在有个周年庆宴会,要求与会职员的快乐指数最大。但是,没有职员愿和直接上司一起与会。
Input
第一行一个整数N。(1<=N<=6000)
接下来N行,第i+1行表示i号职员的快乐指数Ri。(-128<=Ri<=127)
接下来N-1行,每行输入一对整数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
#include<iostream>
#include<cstdio>
using namespace std;
int n,a[6010],zhegebianliangzailuangao[6010];
int tot,hd[6010],head;
int f[6010][2];
struct abc{
int y,to;
}b[6010];
void add(int x,int y)
{
b[++tot].y=y;
b[tot].to=hd[x];
hd[x]=tot;
}
void hh(int now)
{
f[now][1]=a[now];
for(int i=hd[now];i;i=b[i].to)
{
hh(b[i].y);
f[now][1]+=f[b[i].y][0];
f[now][0]+=max(f[b[i].y][1],f[b[i].y][0]);
}
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
int x,y;
for(int i=1;i<n;i++)
{
scanf("%d%d",&x,&y);
add(y,x);
zhegebianliangzailuangao[x]=1;
}
cin>>x>>y;
for(int i=1;i<=n;i++)
if(zhegebianliangzailuangao[i]==0)
{
head=i;
break;
}
hh(head);
cout<<max(f[head][1],f[head][0])<<endl;
}