第十四届蓝桥杯软件赛省赛python b组-试题F 树上选点（树形dp）

试题 F: 树上选点

时间限制: 10.0s

内存限制: 512.0MB

本题总分：15 分

【问题描述】

给定一棵树，树根为 1，每个点的点权为 Vi 。

你需要找出若干个点 Pi，使得：

1. 每两个点 Px Py 互不相邻；

2. 每两个点 Px Py 与树根的距离互不相同；

3. 找出的点的点权之和尽可能大。

请输出找到的这些点的点权和的最大值。

【输入格式】

输入的第一行包含一个整数 n 。

第二行包含 n − 1 个整数 Fi ，相邻整数之间使用一个空格分隔，分别表示

第 2 至 n 个结点的父结点编号。

第三行包含 n 个整数 Vi，相邻整数之间使用一个空格分隔，分别表示每个

结点的点权。

【输出格式】

输出一行包含一个整数表示答案。

【样例输入】

5

1 2 3 2

2 1 9 3 5

【样例输出】

11

n=int(input())
f=list(map(int,input().split()))
w=list(map(int,input().split()))
f.insert(0,0)
f.insert(1,1)
w.insert(0,0)
L=[0]*(n+1)
L[1]=1
dp=w[:]
for i in range(2,n+1):
    L[i]=L[f[i]]+1
for j in range(1,n+1):
    for x in range(1,n+1):
        if L[x]<L[j] and f[j]!=x:
            dp[j]=max(dp[j],dp[x]+w[j])
print(max(dp))
print(dp)