京东算法笔试第一道-及时止损27%
一直显示是27%，报的是WA，后面测试一下后27%的测试案例我输出的答案超过1e8。
我的做法是先给a从小到到大排序
损失的计算公式如下，sum部分做了优化，值存在数组中。
loss[i]=loss[i-m]+sum(A[:i])
有ac 的大佬来分享下正确解法吗？
附上完整代码。
from sys import stdin n, m = map(int, stdin.readline().split()) number = list(map(int, stdin.readline().split())) number.sort() Q = int(input()) number_sum = [0] for item in number: last = number_sum[-1] number_sum.append(last + item) sum_loss = [] for i in range(m): sum_loss.append(number_sum[i + 1]) for i in range(m, n): sum_loss.append(sum_loss[i - m] + number_sum[i + 1]) sum_loss = [0] + sum_loss for i in range(Q): q = int(input()) print(sum_loss[q])
远景公司福利 157人发布