洛谷_P1223 排队接水_python写法

P1223 排队接水 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

n = int(input())
data = list(map(int,input().split()))
data.insert(0,0)

li = []
for i, elem in enumerate(data):
    li.append([i,elem])
li.sort(key=lambda x:x[1])

for i in range(1,n+1):
    print(li[i][0],end=' ')

print('\n',end='')
sum = 0
for i in range(len(li)-1):
    sum += (li[i][1])*(n-i)
av = sum/n
print('%.2f' %av)