问题描述
有一棵 n 个节点的树,树上每个节点都有一个正整数权值。如果一个点被选择了,那么在树上和它相邻的点都不能被选择。求选出的点的权值和最大是多少?
输入格式
第一行包含一个整数 n 。
接下来的一行包含 n 个正整数,第 i 个正整数代表点 i 的权值。
接下来一共 n-1 行,每行描述树上的一条边。
输出格式
输出一个整数,代表选出的点的权值和的最大值。
样例输入
5
1 2 3 4 5
1 2
1 3
2 4
2 5
样例输出
12
样例说明
选择3、4、5号点,权值和为 3+4+5 = 12 。
数据规模与约定
对于20%的数据, n <= 20。
对于50%的数据, n <= 1000。
对于100%的数据, n <= 100000。
权值均为不超过1000的正整数。
题解:树形dp。(点权)
设dp[i][0]表示不选择i点时,i点及其子树能选出的最大权值。
dp[ i ][1]表示选择i点时,i点及其子树的最大权值。
所以,状态转移方程为:
对于叶子节点 i :
dp[ i ][0] = 0, dp[ i ][1] = K.(点权)
对于非叶子节点 i :
dp[ i ][1] = i点权值 + ∑dp[ j ][ 0 ] (j是i的儿子)
dp[ i ][0] = ∑max(dp[ j ][ 0 ], dp[ j ][1]) (j是i的儿子)
最后答案的最大权值即为max(dp[1][0], dp[1][1])。
代码: