没有上司的舞会

来源:互联网 发布:手机定位考勤软件 编辑:程序博客网 时间: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
原创粉丝点击