没有上司的舞会
来源:互联网 发布:手机定位考勤软件 编辑:程序博客网 时间:2024/04/29 07:52
【题目描述】
有N个职员,编号为1~N。他们有从属关系,也就是说他们的关系就像一棵以校长为根的树,父结点就是子结点的直接上司。每个职员有一个快乐指数。现在有个周年庆宴会,要求与会职员的快乐指数最大。但是,没有职员愿和直接上司一起与会。
【输入描述】
第一行一个整数N(1 <= N <= 6000);
接下来N行,第i+1行表示i号职员的快乐指数Ri(-128 <= Ri <= 127);
接下来N-1行,每行输入一对整数L、K,表示K是L的直接上司;
最后一行输入0、0。
【输出描述】
输出最大的快乐指数。
【样例输入】
7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
0 0
【样例输出】
5
源代码:#include<cstdio>#include<algorithm>using namespace std;int n,t1,t2,root=1,f[6001][2]={0},father[6001]={0}; //数组范围!不要吝惜空间!void DP(int t) //树形DP。{ for (int a=1;a<=n;a++) if (father[a]==t) //判断是否为其直属部下。 { DP(a); //递归到叶节点,后回溯求解。 f[t][1]+=f[a][0]; //编号为t的职员若来,则应加上他的直属部下不来的解。 f[t][0]+=max(f[a][1],f[a][0]); //若不来,则在他的直属部下来与不来的解中取最优值。 }}int main(){ scanf("%d",&n); for (int a=1;a<=n;a++) scanf("%d",&f[a][1]); //编号为a的职员来时增加的欢乐度。 scanf("%d%d",&t1,&t2); while (t1) { father[t1]=t2; scanf("%d%d",&t1,&t2); } while (father[root]) //寻找根节点。 root=father[root]; DP(root); printf("%d",max(f[root][1],f[root][0])); return 0;}
0 0