树形dp：没有上司的舞台

没有上司的舞台

---

题目描述
某大学有N个职员，编号为1~N。他们之间有从属关系，也就是说他们的关系就像一棵以校长为根的树，父结点就是子结点的直接上司。现在有个周年庆宴会，宴会每邀请来一个职员都会增加一定的快乐指数Ri，但是呢，如果某个职员的上司来参加舞会了，那么这个职员就无论如何也不肯来参加舞会了。所以，请你编程计算，邀请哪些职员可以使快乐指数最大，求最大的快乐指数。
输入输出格式
输入格式：

第一行一个整数N。(1<=N<=6000)
接下来N行，第i+1行表示i号职员的快乐指数Ri。(-128<=Ri<=127)
接下来N-1行，每行输入一对整数L,K。表示K是L的直接上司。
最后一行输入0 0

输出格式：

输出最大的快乐指数。

输入输出样例
输入样例#1：
7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
0 0
输出样例#1：
5
思路：
我们利用邻接表了存储用户输入的树形成一个树形dp，然后我们先找到主的根节点进行计算，在函数中在变各个子节点。 dp[u][1]=kx[u]表示当u编号人员要来的欢乐指数，然后j=z[k]查找下个的子节点。没有的话就循环u的子节点。
当u号不去时用dp[u][0]来表示，去时用dp[u][1]表示。
当u号不去时dp[u][0]+=max(dp[j][0],dp[j][1]) ，选择他的子节点去和不去的可能的最大值。
当u号去时dp[u][1]+=dp[j][0] ，所以他的子节点不能去。
程序：

def dfs(u):
    global dp
    dp[u][1]=kx[u]
    k=h[u]
    while k!=0:
        j=z[k]
        dfs(j)
        dp[u][0]+=max(dp[j][0],dp[j][1])
        dp[u][1]+=dp[j][0]
        k=nx[k]
    
n=int(input())
kx=[0]
c=[]
for i in range(n):
    kx.append(int(input()))
for i in range(n):
    c.append(list(map(int,input().split())))
c.pop()
nu=1
h=[0 for i in range(n+1)]
nx=[0 for i in range(n+1)]
z=[0 for i in range(n+1)]
dp=[[0 for i in range(2)] for i in range(n+1)]
fa=[0 for i in range(n+1)]
for i in range(n-1):
    fa[c[i][0]]=1
    z[nu]=c[i][0]
    nx[nu]=h[c[i][1]]
    h[c[i][1]]=nu
    nu+=1
for i in range(1,n):
    if fa[i]==0:
        dfs(i)
        s=i
        break
print(max(dp[s][0],dp[s][1]))

禁止转载。仅用于自己学习。对程序错误不负责。