数据分析笔试编程题1

笔试1
题目连接

import math
num1 = input().split(' ')
n = int(num1[0])
m = int(num1[1])
num2 = input().split(' ')
a_i = []
for i in range(n):
    a_i.append(int(num2[i]))
a_i.sort()
Q = int(input())
q = []
for i in range(Q):
    q.append(int(input()))
for i in range(Q):
    temp1 = a_i[:q[i]]
    temp1.sort(reverse=True)
    loss = 0
    for j in range(q[i]):
        num = math.ceil((j+1)/m)
        loss += num*temp1[j]
    print(loss)

你可能感兴趣的:(数据分析)