【题目描述】
有N个职员,编号为1~N,他们有从属关系,就像一棵树,父结点就是子结点的直接上司,每个职员有一个快乐指数。
现有一个舞会,需要与会职员的快乐指数总和最大,但是,没有职员愿意和直接上司一起与会。
【输入描述】
第一行输入一个整数N(1 <= N <= 6000);
接下来N行,每行输入一个整数,表示职员的快乐指数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; }