问题：课程结束时，有 n 个学生向基里尔提出问题，其中他总共有 m 个问题。每个学生都希望得到a题，如果他得到的题少，即b题，他就会生气(a − b)^2怒。每项任务只能分配给一名学生。基里尔真的不想激怒他心爱的学生，所以他请求你帮助减少课堂上的总愤怒。
def angry(m, n, List):
numb = 0
for i in range(m):
List.sort(reverse=True)
List[0] -= 1
for i in range(n):
numb += List[i] ** 2
return numb
m, n = map(int, input().split())
a = list(map(int, input().split()))
# Вывод минимальной злости
print(angry(m, n, a) % (2 ** 64))
也尝试过
def angry(m, n, List):
for i in range(m):
List.sort(reverse=True)
List[0] -= 1
return sum([x ** 2 for x in List])
m, n = map(int, input().split())
a = list(map(int, input().split()))
# Вывод минимальной злости
print(angry(m, n, a) % (2**64))
错误：“超出最长操作时间。 ”
我怎样才能加速这段代码以使最终结果相同？