python蓝桥杯：结点选择（70分）

树形动态规划

自从上次做完K好数动态规划，一直没做树形规划这里。今天花了5个小时，才终于写出一个70分的代码，好垃圾啊…
接下来讲讲我遇到的问题

问题一：树的存储

突然想起来学C的时候树都是有序输入的，但是这里不是
这里其实是图的存储，百度看了一堆，大致是邻接矩阵和邻接表两种，而用的更普遍的好像是邻接表。
于是我就打算用邻接表来存储，但是发现别人做的代码好像都是用c++写的，用vector生成二维数组来存储（对吗？求指正）
我存储是用的字典的key存序号，value是一个list，存储相邻的序号。

table = {}
for i in range(n):
    table.update({i + 1: []})
for i in range(n - 1):
    father, child = list(map(int, input().split()))
    table.get(father).append(child)
    table.get(child).append(father)
#一共有n-1条边，每次输入一条边。
#例如输入5个点
				1 2
				1 3
				2 4
				2 5
存储结构长这样：{1: [2, 3], 2: [1, 4, 5], 3: [1], 4: [2], 5: [2]}

问题二：有了树怎么下手

又是长时间的百度，看基础的树形动态规划，还是不太会，只能看别人的C++代码才能理解这道题咋做。好菜啊。
思想主要是递归，发现树的操作好像都是递归？
对于每个点，有两个选择，分别是选和不选。对于第i个结点，我们用value[i][0]表示不取该结点，所能达到的最大值；value[i][1]表示取该结点，该结点所能达到的最大值。第一种情况，当我不取第i个结点，当前结点所能达到的最大值是，等于i的每个孩子结点取或者不取所能达到的较大值的和；第二种情况，当我取第i个结点，当前结点所能达到的最大值是，等于i的每个孩子不取所能达到的最大值的和。即dp[i][0] += max(dp[child][0],dp[child][1]) 、 dp[i][1] += dp[child][0]

使用深度搜索dfs，自顶向下分解，自底向上回溯求解。假定0是根节点的父亲结点。
————————————————
上面几行话是引用的，我讲述能力太差了。。。。。。
原文链接：https://blog.csdn.net/jaster_wisdom/article/details/79624852

def dfs(node,pre):
    global value,table
    for i in table.get(node):
        if i !=pre:
            dfs(i,node)
            value[node][0]+=max(value[i][0],value[i][1])
            value[node][1]+=value[i][0]

递归真有意思，但是还是不能熟练运用递归。

完整代码

def dfs(node,pre):
    global value,table
    for i in table.get(node):
        if i !=pre:
            dfs(i,node)
            value[node][0]+=max(value[i][0],value[i][1])
            value[node][1]+=value[i][0]

def main():
    global value, table
    n = int(input())
    value = list(map(int, input().split()))
    value = list(map(lambda x:[0,x],value))
    value.insert(0,0)
    table = {}
    for i in range(n):
        table.update({i + 1: []})
    for i in range(n - 1):
        father, child = list(map(int, input().split()))
        table.get(father).append(child)
        table.get(child).append(father)
    print(table)
    dfs(1,0)
    print(max(value[1][0],value[1][1]))

if __name__=='__main__':
    main()

最终运行结果只有70分，最后三个应该是超时了，如果有大佬没超时，或者有更好的方法，跪求告知
。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。