蓝桥杯--结点选择（树形动态规划）

蓝桥杯–结点选择（python，树形动态规划）

问题描述

有一棵 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可以用dfs来实现。

dp[1][i] 表示取当前节点的最大值。

dp[0][i] 表示不去当前节点的最大值。

由于数据是一棵数，所以从任意节点开始dfs都可以。注意标记。

dp[1][u]+=dp[0][v]; //取当前节点，那么加上不取邻接点的dp值

dp[0][u]+=max(dp[0][v],dp[1][v]); //不取当前节点，那么需要加上取或不取邻接点dp值的最大值。

因为需要取一层，所以需要用+=。

代码

n = int(input())
l = list(map(int,input().split()))
l.insert(0,0)
dic = {}
for i in range(n-1):
    a,b = list(map(int,input().split()))
    dic[a] = dic.get(a,[])
    dic[b] = dic.get(b,[])
    dic[a].append(b)
    dic[b].append(a)

vis = [0 for _ in range(n+1)]
dp = [[0 for j in range(n+1)] for i in range(2)]

def dfs(u):
    vis[u] = 1
    for i in range(len(dic[u])):
        v = dic[u][i]
        if vis[v]:
            continue
        dfs(v)
        dp[1][u] += dp[0][v]
        dp[0][u] += max(dp[0][v],dp[1][v])
    dp[1][u] += l[u]

dfs(1)
print(max(dp[0][1],dp[1][1]))