阿里算法笔试

编程第二题想知道错在哪里，怎么都ac不了

import sys

if __name__ == "__main__":

# 读取第一行的n

h = sys.stdin.readline().strip()

n,k = map(int, h.split())

score = sys.stdin.readline().strip()

score = list(map(int, score.split()))

score = [-1000] + score

edge = [[] for _ in range(n+1)]

for i in range(n-1):

# 读取每一行

line = sys.stdin.readline().strip()

# 把每一行的数字分隔后转化成int列表

e = list(map(int, line.split()))

edge[e[0]].append(e[1])

def dfs(x):

if edge[x] == []:

return score[x]

for i in range(len(edge[x])):

score[x] += dfs(edge[x][i])

return score[x]

dfs(1)

score.sort(key=lambda x:-x)

print(score[k-1])